ColossalAI/colossalai/legacy/amp/apex_amp
..
__init__.py
apex_amp.py