html refactor (#555)

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

-colossalai.amp.naive\_amp.grad\_scaler
-======================================
-
-.. automodule:: colossalai.amp.naive_amp.grad_scaler
-   :members:
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.amp.naive_amp.grad_scaler.base_grad_scaler
-   colossalai.amp.naive_amp.grad_scaler.constant_grad_scaler
-   colossalai.amp.naive_amp.grad_scaler.dynamic_grad_scaler

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,14 +3,3 @@ colossalai.amp.naive\_amp
 
 .. automodule:: colossalai.amp.naive_amp
    :members:
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.amp.naive_amp.grad_scaler
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.amp.naive_amp.naive_amp

docs/colossalai/colossalai.amp.rst

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

docs/colossalai/colossalai.amp.torch_amp.rst

@@ -3,9 +3,3 @@ colossalai.amp.torch\_amp
 
 .. automodule:: colossalai.amp.torch_amp
    :members:
\ No newline at end of file
-
-
-.. 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
 ==================
 
-.. automodule:: colossalai.builder
-   :members:
-
+*This module contains different methods to build* ``schedule``, ``model``, ``optimizer``,
+``layers`` *and other important classes.*
 
 .. toctree::
    :maxdepth: 2

docs/colossalai/colossalai.communication.rst

 colossalai.communication
 ========================
 
-.. automodule:: colossalai.communication
-   :members:
-
+*This module contains different methods for communication in parallel training.*
 
 .. 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

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:
+   :members: Initializer_2D_Row, Initializer_2D_Col

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:
+   :members: Initializer_2p5D_ROW, Initializer_2p5D_Col, Initializer_2p5D_Dep, Initializer_2p5D_XZ

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:
+   :members: Initializer_3D_Input, Initializer_3D_Weight, Initializer_3D_Output

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:
+   :members: Initializer_Sequence_DP

docs/colossalai/colossalai.context.process_group_initializer.rst

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

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.gradient_handler.rst

@@ -3,9 +3,3 @@ colossalai.engine.gradient\_handler
 
 .. automodule:: colossalai.engine.gradient_handler
    :members:
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.engine.gradient_handler.utils

docs/colossalai/colossalai.engine.gradient_handler.utils.rst

-colossalai.engine.gradient\_handler.utils
-=========================================
-
-.. automodule:: colossalai.engine.gradient_handler.utils
-   :members: