ColossalAI/colossalai/legacy/zero/init_ctx
..
__init__.py
init_context.py