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

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