colossalai.amp.torch\_amp.torch\_amp
====================================

.. automodule:: colossalai.amp.torch_amp.torch_amp
   :members: