haiscale.pipeline
==========================

.. toctree::
   :maxdepth: 2
   :caption: Contents:

.. py:currentmodule:: haiscale.pipeline
.. autosummary::
   :nosignatures:

   PipeDream
   GPipe
   Interleaved1F1B
   SequentialModel
   partition
   make_subgroups


.. py:currentmodule:: haiscale.pipeline
.. autoclass:: PipeDream
  :members: forward_backward, forward, stage, num_stages

.. py:currentmodule:: haiscale.pipeline
.. autoclass:: Interleaved1F1B
  :members: forward_backward, forward, stage, num_stages

.. autoclass:: GPipe
  :members: forward_backward, stage, num_stages

.. autoclass:: SequentialModel
.. autofunction:: partition
.. autofunction:: make_subgroups