You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai
littsk 11f1e426fe
[hotfix] Correct several erroneous code comments (#4794)
1 year ago
..
_C [setup] support pre-build and jit-build of cuda kernels (#2374) 2 years ago
_analyzer [misc] update pre-commit and run all files (#4752) 1 year ago
amp [shardformer] fix master param sync for hybrid plugin/rewrite unwrapping logic (#4758) 1 year ago
auto_parallel [misc] update pre-commit and run all files (#4752) 1 year ago
autochunk [misc] update pre-commit and run all files (#4752) 1 year ago
booster [lazy] support from_pretrained (#4801) 1 year ago
checkpoint_io [checkpointio] support unsharded checkpointIO for hybrid parallel (#4774) 1 year ago
cli [bug] Fix the version check bug in colossalai run when generating the cmd. (#4713) 1 year ago
cluster [doc] polish shardformer doc (#4779) 1 year ago
context [misc] update pre-commit and run all files (#4752) 1 year ago
device [misc] update pre-commit and run all files (#4752) 1 year ago
fx [misc] update pre-commit and run all files (#4752) 1 year ago
inference [inference] chatglm2 infer demo (#4724) 1 year ago
interface [lazy] support from_pretrained (#4801) 1 year ago
kernel [inference] chatglm2 infer demo (#4724) 1 year ago
lazy [doc] add lazy init docs (#4808) 1 year ago
legacy [bug] fix get_default_parser in examples (#4764) 1 year ago
logging [misc] update pre-commit and run all files (#4752) 1 year ago
nn [misc] add last_epoch in CosineAnnealingWarmupLR (#4778) 1 year ago
pipeline [misc] update pre-commit and run all files (#4752) 1 year ago
shardformer [hotfix] Correct several erroneous code comments (#4794) 1 year ago
tensor [misc] update pre-commit and run all files (#4752) 1 year ago
testing [misc] update pre-commit and run all files (#4752) 1 year ago
utils [misc] update pre-commit and run all files (#4752) 1 year ago
zero [hotfix] Correct several erroneous code comments (#4794) 1 year ago
__init__.py [misc] update pre-commit and run all files (#4752) 1 year ago
initialize.py [misc] update pre-commit and run all files (#4752) 1 year ago