ColossalAI/colossalai/legacy/registry
..
__init__.py
registry.py