ColossalAI/docs/colossalai/colossalai.zero.utils.rst

13 lines
200 B
ReStructuredText

colossalai.zero.utils
=====================
.. automodule:: colossalai.zero.utils
:members:
.. toctree::
:maxdepth: 2
colossalai.zero.utils.zero_hook
colossalai.zero.utils.gemini_hook