ColossalAI/colossalai/legacy/utils/checkpoint
..
__init__.py
module_checkpoint.py
utils.py