index.rst 460 Bytes
Newer Older
Shaden Smith's avatar
Shaden Smith committed
1
2
3
DeepSpeed
=========

Jeff Rasley's avatar
Jeff Rasley committed
4
5
6
Model Setup
-----------

Shaden Smith's avatar
Shaden Smith committed
7
8
9
.. toctree::
   :maxdepth: 2

10
   initialize
Jeff Rasley's avatar
Jeff Rasley committed
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

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

.. toctree::
   :maxdepth: 2

   training


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

   model-checkpointing
   activation-checkpointing
Shaden Smith's avatar
Shaden Smith committed
28
29


30
31
32
33
34
35
36
37
Transformer Kernel API
----------------------
.. toctree::
   :maxdepth: 2

   kernel


Shaden Smith's avatar
Shaden Smith committed
38
Indices and tables
Jeff Rasley's avatar
Jeff Rasley committed
39
------------------
Shaden Smith's avatar
Shaden Smith committed
40
41
42
43

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