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

.. automodule:: colossalai.engine.amp
   :members:


.. toctree::
   :maxdepth: 2

   colossalai.engine.amp.amp_type
   colossalai.engine.amp.grad_scaler