Hongxin Liu ac178ca5c1 [legacy] move builder and registry to legacy (#4603)		2023-09-05 21:53:10 +08:00
..
__init__.py	[zero] reorganize zero/gemini folder structure (#3424)	2023-04-04 13:48:16 +08:00
_utils.py	[bf16] add bf16 support (#3882)	2023-06-05 15:58:31 +08:00
reduce_scatter.py	[zero] reorganize zero/gemini folder structure (#3424)	2023-04-04 13:48:16 +08:00
sharded_model_v2.py	[nfc] fix typo colossalai/zero (#3923)	2023-06-08 00:01:29 +08:00
utils.py	[zero] reorganize zero/gemini folder structure (#3424)	2023-04-04 13:48:16 +08:00
zero_hook.py	[legacy] move builder and registry to legacy (#4603)	2023-09-05 21:53:10 +08:00