ColossalAI/colossalai/zero
Jiarui Fang 6b6002962a [zero] zero init context collect numel of model (#375) 2022-03-11 15:50:28 +08:00
..
init_ctx [zero] zero init context collect numel of model (#375) 2022-03-11 15:50:28 +08:00
shard_utils
sharded_model
sharded_optim
sharded_param
__init__.py