colossalai.nn.loss
==================
.. automodule:: colossalai.nn.loss
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.loss.loss_1d
colossalai.nn.loss.loss_2d
colossalai.nn.loss.loss_2p5d
colossalai.nn.loss.loss_3d
colossalai.nn.loss.loss_moe