Jiarui Fang 67b4928244 [zero] polish init context (#645)

2022-04-02 15:52:04 +08:00

..

init_ctx

[zero] polish init context (#645)

2022-04-02 15:52:04 +08:00

shard_utils

[refactor] memory utils (#577)

2022-04-01 09:22:33 +08:00

sharded_model

polish docstring of zero (#612)

2022-04-01 14:50:56 +08:00

sharded_optim

[zero] adapt zero for unsharded paramters (Optimizer part) (#601)

2022-04-01 20:10:47 +08:00

sharded_param

[refactor] memory utils (#577)

2022-04-01 09:22:33 +08:00

__init__.py

[refactor] remove old zero code (#517)

2022-03-25 14:54:39 +08:00