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