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.

digger yu 9265f2d4d7 [NFC]fix typo colossalai/auto_parallel nn utils etc. (#3779) ... * fix typo colossalai/autochunk auto_parallel amp * fix typo colossalai/auto_parallel nn utils etc.		2 years ago
..
chunk	[NFC]fix typo colossalai/auto_parallel nn utils etc. (#3779)	2 years ago
memory_tracer	[NFC]fix typo colossalai/auto_parallel nn utils etc. (#3779)	2 years ago
__init__.py	…
colo_init_context.py	…
gemini_ddp.py	…
gemini_hook.py	…
gemini_mgr.py	…
gemini_optimizer.py	…
placement_policy.py	…
utils.py	…