ver217 001ca624dd impl shard optim v2 and add unit test		2022-03-11 15:50:28 +08:00
..
shard_utils	[zero] a shard strategy in granularity of tensor (#307)	2022-03-11 15:50:28 +08:00
sharded_model	add sharded adam	2022-03-11 15:50:28 +08:00
sharded_optim	impl shard optim v2 and add unit test	2022-03-11 15:50:28 +08:00
sharded_param	[zero] sharded tensor (#305)	2022-03-11 15:50:28 +08:00
__init__.py	added buffer sync to naive amp model wrapper (#291)	2022-03-11 15:50:28 +08:00