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.

Jiarui Fang 214da761d4 [zero] add stateful tensor (#549)		3 years ago
..
init_ctx	[zero] add zero context manager to change config during initialization (#546)	3 years ago
shard_utils	[zero] non model data tracing (#545)	3 years ago
sharded_model	[zero] add stateful tensor (#549)	3 years ago
sharded_optim	[zero] dump memory stats for sharded model (#548)	3 years ago
sharded_param	[zero] add stateful tensor (#549)	3 years ago
__init__.py	[refactor] remove old zero code (#517)	3 years ago