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.

Baizhou Zhang 58913441a1 Next commit [checkpointio] Unsharded Optimizer Checkpoint for Gemini Plugin (#4141) ... * [checkpointio] unsharded optimizer checkpoint for Gemini plugin * [checkpointio] unsharded optimizer checkpoint for Gemini using all_gather		1 year ago
..
__init__.py	[shardformer] supported T5 and its variants (#4045)	1 year ago
comparison.py	Next commit [checkpointio] Unsharded Optimizer Checkpoint for Gemini Plugin (#4141)	1 year ago
pytest_wrapper.py	[testing] move pytest to be inside the function (#4087)	1 year ago
random.py	[zero] test gradient accumulation (#1964)	2 years ago
utils.py	[booster] support torch fsdp plugin in booster (#3697)	2 years ago