ColossalAI/docs/colossalai/colossalai.utils.rst

28 lines
543 B
ReStructuredText

colossalai.utils
================
.. automodule:: colossalai.utils
:members:
.. toctree::
:maxdepth: 2
colossalai.utils.checkpoint
colossalai.utils.data_sampler
colossalai.utils.model
colossalai.utils.multi_tensor_apply
colossalai.utils.profiler
colossalai.utils.tensor_detector
.. toctree::
:maxdepth: 2
colossalai.utils.activation_checkpoint
colossalai.utils.checkpointing
colossalai.utils.common
colossalai.utils.cuda
colossalai.utils.memory
colossalai.utils.moe
colossalai.utils.timer