colossalai.nn
=============

.. automodule:: colossalai.nn
   :members:

.. toctree::
   :maxdepth: 2

   colossalai.nn.layer
   colossalai.nn.loss
   colossalai.nn.lr_scheduler
   colossalai.nn.metric
   colossalai.nn.model
   colossalai.nn.optimizer


.. toctree::
   :maxdepth: 2

   colossalai.nn.init
