ColossalAI/colossalai/zero
Hongxin Liu 5ff5323538
[hotfix] fix zero optim save (#6191)
2025-02-14 15:09:50 +08:00
..
gemini [checkpointio]support asyncio for 3d (#6152) 2024-12-23 10:24:22 +08:00
low_level [hotfix] fix zero optim save (#6191) 2025-02-14 15:09:50 +08:00
__init__.py [shardformer]: support gpt-j, falcon, Mistral and add interleaved pipeline for bert (#5088) 2023-11-28 16:54:42 +08:00
wrapper.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00