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

.. automodule:: colossalai.amp
   :members:

.. toctree::
   :maxdepth: 2

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


.. toctree::
   :maxdepth: 2

   colossalai.amp.amp_type
