colossalai.utils.model
======================

.. automodule:: colossalai.utils.model
   :members:


.. toctree::
   :maxdepth: 2

   colossalai.utils.model.colo_init_context
   colossalai.utils.model.lazy_init_context
   colossalai.utils.model.utils
