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
wukong1992 b37797ed3d
[booster] support torch fsdp plugin in booster (#3697)
2 years ago
..
_C
_analyzer [example] add train resnet/vit with booster example (#3694) 2 years ago
amp
auto_parallel [NFC] fix typo applications/ and colossalai/ (#3735) 2 years ago
autochunk [NFC] fix typo applications/ and colossalai/ (#3735) 2 years ago
booster [booster] support torch fsdp plugin in booster (#3697) 2 years ago
builder
checkpoint_io [NFC] fix typo applications/ and colossalai/ (#3735) 2 years ago
cli [NFC] fix typo applications/ and colossalai/ (#3735) 2 years ago
cluster
communication [CI] fix some spelling errors (#3707) 2 years ago
context [CI] fix some spelling errors (#3707) 2 years ago
device
engine
fx [doc] Fix typo under colossalai and doc(#3618) 2 years ago
interface
kernel [doc] Fix typo under colossalai and doc(#3618) 2 years ago
logging
nn [doc] Fix typo under colossalai and doc(#3618) 2 years ago
pipeline
registry
tensor [tensor] Refactor handle_trans_spec in DistSpecManager 2 years ago
testing [booster] support torch fsdp plugin in booster (#3697) 2 years ago
trainer
utils [doc] Fix typo under colossalai and doc(#3618) 2 years ago
zero [booster] gemini plugin support shard checkpoint (#3610) 2 years ago
__init__.py
constants.py
core.py
global_variables.py
initialize.py