haiscale.fsdp
======================

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

.. py:currentmodule:: haiscale.fsdp
.. autosummary::
   :nosignatures:

   FullyShardedDataParallel


.. py:currentmodule:: haiscale.fsdp
.. autoclass:: FullyShardedDataParallel
   :members: summon_full_params