ColossalAI/colossalai/amp
..
apex_amp
naive_amp
torch_amp
__init__.py
amp_type.py