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)
1 year ago
..
gemini fix typo with colossalai/trainer utils zero (#3908) 1 year ago
init_ctx [nfc] fix typo colossalai/zero (#3923) 1 year ago
shard_utils
sharded_model [nfc] fix typo colossalai/zero (#3923) 1 year ago
sharded_optim [bf16] add bf16 support (#3882) 2 years ago
sharded_param
__init__.py