ColossalAI/docs/colossalai/colossalai.nn.layer.moe.rst

colossalai.nn.layer.moe
=======================

.. automodule:: colossalai.nn.layer.moe
   :members:


.. toctree::
   :maxdepth: 2

   colossalai.nn.layer.moe.experts
   colossalai.nn.layer.moe.layers
   colossalai.nn.layer.moe.utils
removed tutorial markdown and refreshed rst files for consistency 2022-01-19 08:06:53 +00:00			`colossalai.nn.layer.moe`
			`=======================`

			`.. automodule:: colossalai.nn.layer.moe`
			`:members:`
update rst (#615) 2022-04-01 07:46:38 +00:00

			`.. toctree::`
			`:maxdepth: 2`

			`colossalai.nn.layer.moe.experts`
			`colossalai.nn.layer.moe.layers`
			`colossalai.nn.layer.moe.utils`