DeepSpeed
=========

Model Setup
-----------

.. toctree::
   :maxdepth: 2

   initialize

Training API
------------

.. toctree::
   :maxdepth: 2

   training


Checkpointing API
-----------------
.. toctree::
   :maxdepth: 2

   model-checkpointing
   activation-checkpointing


Transformer Kernel API
----------------------
.. toctree::
   :maxdepth: 2

   kernel


Indices and tables
------------------

* :ref:`genindex`
* :ref:`modindex`
* :ref:`search`
