.. py:module:: megengine.amp
.. currentmodule:: megengine.amp

=============
megengine.amp
=============
.. autosummary::
   :toctree: api
   :nosignatures:

   GradScaler
   autocast
   convert_module_format
   convert_tensor_format

.. autosummary::
   :toctree: api
   :nosignatures:
   :template: autosummary/mprop.rst

   enabled
   high_prec_dtype
   low_prec_dtype