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
Hongxin Liu b5f9e37c70
[legacy] clean up legacy code (#4743)
1 year ago
..
_C [setup] support pre-build and jit-build of cuda kernels (#2374) 2 years ago
_analyzer [example] add train resnet/vit with booster example (#3694) 2 years ago
amp [legacy] clean up legacy code (#4743) 1 year ago
auto_parallel [legacy] clean up legacy code (#4743) 1 year ago
autochunk fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808) 2 years ago
booster [shardformer] add custom policy in hybrid parallel plugin (#4718) 1 year ago
checkpoint_io [legacy] clean up legacy code (#4743) 1 year ago
cli [legacy] clean up legacy code (#4743) 1 year ago
cluster [shardformer] support interleaved pipeline (#4448) 1 year ago
context [legacy] clean up legacy code (#4743) 1 year ago
device fix some typo with colossalai/device colossalai/tensor/ etc. (#4171) 1 year ago
fx [legacy] clean up legacy code (#4743) 1 year ago
inference [hotfix] Fix import error: colossal.kernel without triton installed (#4722) 1 year ago
interface [zero] hotfix master param sync (#4618) 1 year ago
kernel [kernel] update triton init #4740 (#4740) 1 year ago
lazy [shardformer] support lazy init (#4202) 1 year ago
legacy [legacy] clean up legacy code (#4743) 1 year ago
logging [legacy] clean up legacy code (#4743) 1 year ago
nn [legacy] clean up legacy code (#4743) 1 year ago
pipeline [legacy] clean up legacy code (#4743) 1 year ago
shardformer [doc] Add user document for Shardformer (#4702) 1 year ago
tensor [legacy] clean up legacy code (#4743) 1 year ago
testing Next commit [checkpointio] Unsharded Optimizer Checkpoint for Gemini Plugin (#4141) 1 year ago
utils [legacy] clean up legacy code (#4743) 1 year ago
zero [legacy] clean up legacy code (#4743) 1 year ago
__init__.py [legacy] clean up legacy code (#4743) 1 year ago
initialize.py [legacy] clean up legacy code (#4743) 1 year ago