colossalai.amp
==============

*This module contains different amp models and optimizers.*

.. toctree::
   :maxdepth: 2

   colossalai.amp.apex_amp
   colossalai.amp.naive_amp
   colossalai.amp.torch_amp


.. toctree::
   :maxdepth: 2

   colossalai.amp.amp_type

.. automodule:: colossalai.amp
   :members: convert_to_amp