colossalai
==========

.. automodule:: colossalai
   :members:

.. toctree::
   :maxdepth: 2

   colossalai.amp
   colossalai.builder
   colossalai.communication
   colossalai.context
   colossalai.engine
   colossalai.kernel
   colossalai.logging
   colossalai.nn
   colossalai.registry
   colossalai.trainer
   colossalai.utils
   colossalai.zero


.. toctree::
   :maxdepth: 2

   colossalai.initialize
