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/booster/plugin
wukong1992 3229f93e30
[booster] add warning for torch fsdp plugin doc (#3833)
2 years ago
..
__init__.py
dp_plugin_base.py
gemini_plugin.py fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808) 2 years ago
low_level_zero_plugin.py [plugin] a workaround for zero plugins' optimizer checkpoint (#3780) 2 years ago
plugin_base.py
torch_ddp_plugin.py
torch_fsdp_plugin.py [booster] add warning for torch fsdp plugin doc (#3833) 2 years ago