update rst (#615)

docs/colossalai/colossalai.amp.apex_amp.apex_amp.rst

+colossalai.amp.apex\_amp.apex\_amp
+==================================
+
+.. automodule:: colossalai.amp.apex_amp.apex_amp
+   :members:

docs/colossalai/colossalai.amp.apex_amp.rst

@@ -3,3 +3,9 @@ colossalai.amp.apex\_amp
 
 .. automodule:: colossalai.amp.apex_amp
    :members:
+
+
+.. toctree::
+   :maxdepth: 2
+
+   colossalai.amp.apex_amp.apex_amp

docs/colossalai/colossalai.amp.naive_amp.naive_amp.rst

+colossalai.amp.naive\_amp.naive\_amp
+====================================
+
+.. automodule:: colossalai.amp.naive_amp.naive_amp
+   :members:

docs/colossalai/colossalai.amp.naive_amp.rst

@@ -3,3 +3,9 @@ colossalai.amp.naive\_amp
 
 .. automodule:: colossalai.amp.naive_amp
    :members:
+
+
+.. toctree::
+   :maxdepth: 2
+
+   colossalai.amp.naive_amp.naive_amp

docs/colossalai/colossalai.amp.rst

 colossalai.amp
 ==============
 
-*This module contains different amp models and optimizers.*
+.. automodule:: colossalai.amp
+   :members:
 
 .. toctree::
    :maxdepth: 2
@@ -15,6 +16,3 @@ colossalai.amp
    :maxdepth: 2
 
    colossalai.amp.amp_type
-
-.. automodule:: colossalai.amp
-   :members: convert_to_amp

docs/colossalai/colossalai.amp.torch_amp.rst

@@ -2,4 +2,10 @@ colossalai.amp.torch\_amp
 =========================
 
 .. automodule:: colossalai.amp.torch_amp
-   :members:
\ No newline at end of file
+   :members:
+
+
+.. toctree::
+   :maxdepth: 2
+
+   colossalai.amp.torch_amp.torch_amp

docs/colossalai/colossalai.amp.torch_amp.torch_amp.rst

+colossalai.amp.torch\_amp.torch\_amp
+====================================
+
+.. automodule:: colossalai.amp.torch_amp.torch_amp
+   :members:

docs/colossalai/colossalai.builder.rst

 colossalai.builder
 ==================
 
-*This module contains different methods to build* ``schedule``, ``model``, ``optimizer``,
-``layers`` *and other important classes.*
+.. automodule:: colossalai.builder
+   :members:
+
 
 .. toctree::
    :maxdepth: 2

docs/colossalai/colossalai.communication.rst

 colossalai.communication
 ========================
 
-*This module contains different methods for communication in parallel training.*
+.. automodule:: colossalai.communication
+   :members:
+
 
 .. toctree::
    :maxdepth: 2

docs/colossalai/colossalai.constants.rst

+colossalai.constants
+====================
+
+.. automodule:: colossalai.constants
+   :members:

docs/colossalai/colossalai.context.moe_context.rst

 colossalai.context.moe\_context
-====================================
+===============================
 
 .. automodule:: colossalai.context.moe_context
-   :members:
\ No newline at end of file
+   :members:

docs/colossalai/colossalai.context.process_group_initializer.initializer_2d.rst

@@ -2,4 +2,4 @@ colossalai.context.process\_group\_initializer.initializer\_2d
 ==============================================================
 
 .. automodule:: colossalai.context.process_group_initializer.initializer_2d
-   :members: Initializer_2D_Row, Initializer_2D_Col
+   :members:

docs/colossalai/colossalai.context.process_group_initializer.initializer_2p5d.rst

@@ -2,4 +2,4 @@ colossalai.context.process\_group\_initializer.initializer\_2p5d
 ================================================================
 
 .. automodule:: colossalai.context.process_group_initializer.initializer_2p5d
-   :members: Initializer_2p5D_ROW, Initializer_2p5D_Col, Initializer_2p5D_Dep, Initializer_2p5D_XZ
+   :members:

docs/colossalai/colossalai.context.process_group_initializer.initializer_3d.rst

@@ -2,4 +2,4 @@ colossalai.context.process\_group\_initializer.initializer\_3d
 ==============================================================
 
 .. automodule:: colossalai.context.process_group_initializer.initializer_3d
-   :members: Initializer_3D_Input, Initializer_3D_Weight, Initializer_3D_Output
+   :members:

docs/colossalai/colossalai.context.process_group_initializer.initializer_sequence.rst

@@ -2,4 +2,4 @@ colossalai.context.process\_group\_initializer.initializer\_sequence
 ====================================================================
 
 .. automodule:: colossalai.context.process_group_initializer.initializer_sequence
-   :members: Initializer_Sequence_DP
+   :members:

docs/colossalai/colossalai.context.process_group_initializer.rst

@@ -14,7 +14,6 @@ colossalai.context.process\_group\_initializer
    colossalai.context.process_group_initializer.initializer_3d
    colossalai.context.process_group_initializer.initializer_data
    colossalai.context.process_group_initializer.initializer_model
-   colossalai.context.process_group_initializer.initializer_moe
    colossalai.context.process_group_initializer.initializer_pipeline
    colossalai.context.process_group_initializer.initializer_sequence
    colossalai.context.process_group_initializer.initializer_tensor

docs/colossalai/colossalai.context.rst

 colossalai.context
 ==================
 
-*This module is serving for receiving and passing user's configuration to all devices to
-initialize and construct parallel training.*
+.. automodule:: colossalai.context
+   :members:
 
 .. toctree::
    :maxdepth: 2
@@ -15,9 +15,7 @@ initialize and construct parallel training.*
    :maxdepth: 2
 
    colossalai.context.config
-   colossalai.context.parallel_context
    colossalai.context.moe_context
+   colossalai.context.parallel_context
    colossalai.context.parallel_mode
-
-.. automodule:: colossalai.context
-   :members:
+   colossalai.context.singleton_meta

docs/colossalai/colossalai.context.singleton_meta.rst

+colossalai.context.singleton\_meta
+==================================
+
+.. automodule:: colossalai.context.singleton_meta
+   :members:

docs/colossalai/colossalai.core.rst

+colossalai.core
+===============
+
+.. automodule:: colossalai.core
+   :members:

docs/colossalai/colossalai.engine.ophooks.rst

 colossalai.engine.ophooks
-===========================
+=========================
 
 .. automodule:: colossalai.engine.ophooks
    :members:
+
+
+.. toctree::
+   :maxdepth: 2
+
+   colossalai.engine.ophooks.zero_hook