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

.. automodule:: colossalai.engine
   :members:

.. toctree::
   :maxdepth: 2

   colossalai.engine.gradient_handler
   colossalai.engine.schedule


.. toctree::
   :maxdepth: 2

   colossalai.engine.amp_type