Unverified Commit 9e3b8b7a authored by Frank Lee's avatar Frank Lee Committed by GitHub
Browse files

[doc] removed read-the-docs (#2932)

parent 77b88a38
colossalai.nn.parallel.data\_parallel
=====================================
.. automodule:: colossalai.nn.parallel.data_parallel
:members:
colossalai.nn.parallel.layers.colo\_module
==========================================
.. automodule:: colossalai.nn.parallel.layers.colo_module
:members:
colossalai.nn.parallel.layers.embedding
=======================================
.. automodule:: colossalai.nn.parallel.layers.embedding
:members:
colossalai.nn.parallel.layers.linear
====================================
.. automodule:: colossalai.nn.parallel.layers.linear
:members:
colossalai.nn.parallel.layers.module\_utils
===========================================
.. automodule:: colossalai.nn.parallel.layers.module_utils
:members:
colossalai.nn.parallel.layers
=============================
.. automodule:: colossalai.nn.parallel.layers
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.parallel.layers.colo_module
colossalai.nn.parallel.layers.embedding
colossalai.nn.parallel.layers.linear
colossalai.nn.parallel.layers.module_utils
colossalai.nn.parallel.reducer
==============================
.. automodule:: colossalai.nn.parallel.reducer
:members:
colossalai.nn.parallel
======================
.. automodule:: colossalai.nn.parallel
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.parallel.layers
.. toctree::
:maxdepth: 2
colossalai.nn.parallel.data_parallel
colossalai.nn.parallel.reducer
colossalai.nn
=============
.. automodule:: colossalai.nn
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.graph
colossalai.nn.layer
colossalai.nn.loss
colossalai.nn.lr_scheduler
colossalai.nn.metric
colossalai.nn.optimizer
colossalai.nn.parallel
.. toctree::
:maxdepth: 2
colossalai.nn.init
colossalai.pipeline.layer\_sepc
===============================
.. automodule:: colossalai.pipeline.layer_spec
:members:
colossalai.pipeline.pipelinable
===============================
.. automodule:: colossalai.pipeline.pipelinable
:members:
colossalai.pipeline
===================
.. automodule:: colossalai.pipeline
:members:
.. toctree::
:maxdepth: 2
colossalai.pipeline.layer_spec
colossalai.pipeline.pipelinable
colossalai.pipeline.utils
colossalai.pipeline.utils
=========================
.. automodule:: colossalai.pipeline.utils
:members:
colossalai.registry.registry
============================
.. automodule:: colossalai.registry.registry
:members:
colossalai.registry
===================
.. automodule:: colossalai.registry
:members:
.. toctree::
:maxdepth: 2
colossalai.registry.registry
colossalai
==========
.. automodule:: colossalai
:members:
.. toctree::
:maxdepth: 2
colossalai.amp
colossalai.builder
colossalai.cli
colossalai.communication
colossalai.context
colossalai.engine
colossalai.fx
colossalai.gemini
colossalai.kernel
colossalai.logging
colossalai.nn
colossalai.pipeline
colossalai.registry
colossalai.tensor
colossalai.testing
colossalai.trainer
colossalai.utils
colossalai.zero
.. toctree::
:maxdepth: 2
colossalai.constants
colossalai.core
colossalai.global_variables
colossalai.initialize
colossalai.tensor.colo\_parameter
=================================
.. automodule:: colossalai.tensor.colo_parameter
:members:
colossalai.tensor.colo\_tensor
==============================
.. automodule:: colossalai.tensor.colo_tensor
:members:
colossalai.tensor.compute\_spec
===============================
.. automodule:: colossalai.tensor.compute_spec
:members:
colossalai.tensor.const
=======================
.. automodule:: colossalai.tensor.const
:members:
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment