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