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