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

14 lines
236 B
ReStructuredText

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