colossalai.rst

colossalai
==========

.. automodule:: colossalai
   :members:

.. toctree::
   :maxdepth: 2

   colossalai.builder
   colossalai.communication
   colossalai.context
   colossalai.engine
   colossalai.logging
   colossalai.nn
   colossalai.registry
   colossalai.trainer
   colossalai.utils


.. toctree::
   :maxdepth: 2

   colossalai.checkpointing
   colossalai.constants
   colossalai.core
   colossalai.initialize