colossalai.zero.utils
=====================

.. automodule:: colossalai.zero.utils
   :members:


.. toctree::
   :maxdepth: 2

   colossalai.zero.utils.zero_hook
   colossalai.zero.utils.gemini_hook