colossalai.amp.apex\_amp
========================

.. automodule:: colossalai.amp.apex_amp
   :members: