ver217 5cc84d94dc rename shared adam to sharded optim v2

2022-03-03 16:20:34 +08:00

..

sharded_model

add sharded adam

2022-03-03 16:20:34 +08:00

sharded_optim

rename shared adam to sharded optim v2

2022-03-03 16:20:34 +08:00

sharded_param

Polish sharded parameter (#297)

2022-03-03 12:42:57 +08:00

__init__.py

added buffer sync to naive amp model wrapper (#291)

2022-03-02 16:47:17 +08:00