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
ver217 0ef8819c67
polish docstring of zero (#612)
3 years ago
..
init_ctx [zero] adapt zero for unsharded parameters (#561) 3 years ago
shard_utils [refactor] memory utils (#577) 3 years ago
sharded_model polish docstring of zero (#612) 3 years ago
sharded_optim polish docstring of zero (#612) 3 years ago
sharded_param [refactor] memory utils (#577) 3 years ago
__init__.py [refactor] remove old zero code (#517) 3 years ago