Hongxin Liu 63ecafb1fb [checkpointio] optimize zero optim checkpoint io (#4591) ...

* [zero] update checkpoint io to save memory

* [checkpointio] add device map to save memory

2023-09-04 11:26:45 +08:00

..

gemini

[gemini] improve compatibility and add static placement policy (#4479)

2023-08-24 09:29:25 +08:00

legacy

[nfc] fix typo colossalai/zero (#3923)

2023-06-08 00:01:29 +08:00

low_level

[checkpointio] optimize zero optim checkpoint io (#4591)

2023-09-04 11:26:45 +08:00

__init__.py

[gemini] improve compatibility and add static placement policy (#4479)

2023-08-24 09:29:25 +08:00

wrapper.py

[gemini] improve compatibility and add static placement policy (#4479)

2023-08-24 09:29:25 +08:00