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/zero/legacy
digger yu de0d7df33f
[nfc] fix typo colossalai/zero (#3923)
2 years ago
..
gemini fix typo with colossalai/trainer utils zero (#3908) 2 years ago
init_ctx [nfc] fix typo colossalai/zero (#3923) 2 years ago
shard_utils [doc] Fix typo under colossalai and doc(#3618) 2 years ago
sharded_model [nfc] fix typo colossalai/zero (#3923) 2 years ago
sharded_optim [bf16] add bf16 support (#3882) 2 years ago
sharded_param [zero] reorganize zero/gemini folder structure (#3424) 2 years ago
__init__.py [example] update examples related to zero/gemini (#3431) 2 years ago