colossalai.zero.sharded\_model
==============================


colossalai.zero.sharded\_model.sharded\_model\_v2
--------------------------------------------------

.. automodule:: colossalai.zero.sharded_model.sharded_model_v2
   :members:

colossalai.zero.sharded\_model.reduce\_scatter
-----------------------------------------------

.. automodule:: colossalai.zero.sharded_model.reduce_scatter
   :members:

colossalai.zero.sharded\_model.utils
--------------------------------------

.. automodule:: colossalai.zero.sharded_model.utils
   :members:

colossalai.zero.sharded\_model.sharded\_model
--------------------------------------------------

.. automodule:: colossalai.zero.sharded_model
   :members:

colossalai.zero.sharded\_model.sharded\_grad
--------------------------------------------------

.. automodule:: colossalai.zero.sharded_grad
   :members:

colossalai.zero.sharded\_model.param\_manager
----------------------------------------------

.. automodule:: colossalai.zero.sharded_model.param_manager
   :members: