You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/docs/colossalai/colossalai.rst

37 lines
602 B

colossalai
3 years ago
==========
.. automodule:: colossalai
:members:
3 years ago
.. toctree::
:maxdepth: 2
colossalai.amp
3 years ago
colossalai.builder
colossalai.cli
3 years ago
colossalai.communication
colossalai.context
colossalai.engine
colossalai.fx
colossalai.gemini
colossalai.kernel
3 years ago
colossalai.logging
colossalai.nn
colossalai.pipeline
3 years ago
colossalai.registry
colossalai.tensor
colossalai.testing
3 years ago
colossalai.trainer
colossalai.utils
colossalai.zero
3 years ago
.. toctree::
:maxdepth: 2
colossalai.constants
colossalai.core
colossalai.global_variables
colossalai.initialize