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
Haichen Huang dbb9659099
[elixir] add elixir plugin and its unit test (#3865)
2 years ago
..
__init__.py [elixir] add elixir plugin and its unit test (#3865) 2 years ago
dp_plugin_base.py
elixir_plugin.py [elixir] add elixir plugin and its unit test (#3865) 2 years ago
gemini_plugin.py fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808) 2 years ago
low_level_zero_plugin.py
plugin_base.py
torch_ddp_plugin.py
torch_fsdp_plugin.py