Hongxin Liu df63564184 [gemini] support amp o3 for gemini (#4872) ...

* [gemini] support no reuse fp16 chunk

* [gemini] support no master weight for optim

* [gemini] support no master weight for gemini ddp

* [test] update gemini tests

* [test] update gemini tests

* [plugin] update gemini plugin

* [test] fix gemini checkpointio test

* [test] fix gemini checkpoint io

2023-10-12 10:39:08 +08:00

..

gemini

[gemini] support amp o3 for gemini (#4872)

2023-10-12 10:39:08 +08:00

low_level

[hotfix] Correct several erroneous code comments (#4794)

2023-09-27 10:43:03 +08:00

__init__.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

wrapper.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00