html refactor (#555)

docs/colossalai/colossalai.nn.layer.parallel_2p5d.layers.rst

-colossalai.nn.layer.parallel\_2p5d.layers
-=========================================
-
-.. automodule:: colossalai.nn.layer.parallel_2p5d.layers
-   :members:

docs/colossalai/colossalai.nn.layer.parallel_2p5d.rst

@@ -3,9 +3,3 @@ colossalai.nn.layer.parallel\_2p5d
 
 .. automodule:: colossalai.nn.layer.parallel_2p5d
    :members:
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.nn.layer.parallel_2p5d.layers

docs/colossalai/colossalai.nn.layer.parallel_3d.layers.rst

-colossalai.nn.layer.parallel\_3d.layers
-=======================================
-
-.. automodule:: colossalai.nn.layer.parallel_3d.layers
-   :members:

docs/colossalai/colossalai.nn.layer.parallel_3d.rst

@@ -3,9 +3,3 @@ colossalai.nn.layer.parallel\_3d
 
 .. automodule:: colossalai.nn.layer.parallel_3d
    :members:
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.nn.layer.parallel_3d.layers

docs/colossalai/colossalai.nn.layer.parallel_sequence.layers.rst

-colossalai.nn.layer.parallel\_sequence.layers
-=============================================
-
-.. automodule:: colossalai.nn.layer.parallel_sequence.layers
-   :members:

docs/colossalai/colossalai.nn.layer.parallel_sequence.rst

@@ -3,9 +3,3 @@ colossalai.nn.layer.parallel\_sequence
 
 .. automodule:: colossalai.nn.layer.parallel_sequence
    :members:
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.nn.layer.parallel_sequence.layers

docs/colossalai/colossalai.nn.layer.utils.common.rst

-colossalai.nn.layer.utils.common
-================================
-
-.. automodule:: colossalai.nn.layer.utils.common
-   :members:

docs/colossalai/colossalai.nn.layer.utils.rst

@@ -3,9 +3,3 @@ colossalai.nn.layer.utils
 
 .. automodule:: colossalai.nn.layer.utils
    :members:
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.nn.layer.utils.common

docs/colossalai/colossalai.nn.layer.vanilla.layers.rst

-colossalai.nn.layer.vanilla.layers
-==================================
-
-.. automodule:: colossalai.nn.layer.vanilla.layers
-   :members:

docs/colossalai/colossalai.nn.layer.vanilla.rst

@@ -3,9 +3,3 @@ colossalai.nn.layer.vanilla
 
 .. automodule:: colossalai.nn.layer.vanilla
    :members:
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.nn.layer.vanilla.layers

docs/colossalai/colossalai.nn.layer.wrapper.lambda_wrapper.rst

-colossalai.nn.layer.wrapper.lambda\_wrapper
-===========================================
-
-.. automodule:: colossalai.nn.layer.wrapper.lambda_wrapper
-   :members:

docs/colossalai/colossalai.nn.layer.wrapper.pipeline_wrapper.rst

-colossalai.nn.layer.wrapper.pipeline\_wrapper
-=============================================
-
-.. automodule:: colossalai.nn.layer.wrapper.pipeline_wrapper
-   :members:

docs/colossalai/colossalai.nn.layer.wrapper.rst

@@ -3,10 +3,3 @@ colossalai.nn.layer.wrapper
 
 .. automodule:: colossalai.nn.layer.wrapper
    :members:
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.nn.layer.wrapper.lambda_wrapper
-   colossalai.nn.layer.wrapper.pipeline_wrapper

docs/colossalai/colossalai.nn.lr_scheduler.rst

 colossalai.nn.lr\_scheduler
 ===========================
 
-.. automodule:: colossalai.nn.lr_scheduler
-   :members:
-
-
 .. toctree::
    :maxdepth: 2
 

docs/colossalai/colossalai.nn.optimizer.rst

 colossalai.nn.optimizer
 =======================
 
-.. automodule:: colossalai.nn.optimizer
-   :members:
-
-
 .. toctree::
    :maxdepth: 2
 

docs/colossalai/colossalai.nn.rst

 colossalai.nn
 =============
 
-.. automodule:: colossalai.nn
-   :members:
+*This part contains different colossalai layers for constructing your model.
+You can easily use them as the way of using layers in torch.nn.*
+
+*Now colossalai support layer types below:* ``Linear``, ``Classifier``, ``Embedding``,
+``PatchEmbedding``, ``LayerNorm``, ``Dropout`` *for different parallelisms.*
 
 .. toctree::
    :maxdepth: 2
 
    colossalai.nn.layer
+
+*This part contains different loss functions for different parallelisms.*
+
+.. toctree::
+   :maxdepth: 2
+
    colossalai.nn.loss
+
+*This part contains different learning rate schedules to control your learning rate
+in training process*
+
+.. toctree::
+   :maxdepth: 2
+
    colossalai.nn.lr_scheduler
+
+*This part contains different metric to measure performance of your model.*
+
+.. toctree::
+   :maxdepth: 2
+
    colossalai.nn.metric
+
+*This part contains some colossalai optimizers*
+
+.. toctree::
+   :maxdepth: 2
+
    colossalai.nn.model
+
+.. toctree::
+   :maxdepth: 2
+
    colossalai.nn.optimizer
 
+*This part contains different methods to initialize weights.*
 
 .. toctree::
    :maxdepth: 2
 
    colossalai.nn.init
+
+.. automodule:: colossalai.nn
+   :members:

docs/colossalai/colossalai.rst

-colossalai
+Colossal-AI
 ==========
 
-.. automodule:: colossalai
-   :members:
-
 .. toctree::
    :maxdepth: 2
 
@@ -25,5 +22,7 @@ colossalai
 .. toctree::
    :maxdepth: 2
 
-   colossalai.global_variables
    colossalai.initialize
+
+.. automodule:: colossalai
+   :members:

docs/colossalai/colossalai.testing.comparison.rst

-colossalai.testing.comparison
-=============================
-
-.. automodule:: colossalai.testing.comparison
-   :members:

docs/colossalai/colossalai.testing.rst

 colossalai.testing
-==================
+========================
 
 .. automodule:: colossalai.testing
    :members:
\ No newline at end of file
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.testing.comparison
-   colossalai.testing.utils

docs/colossalai/colossalai.testing.utils.rst

-colossalai.testing.utils
-========================
-
-.. automodule:: colossalai.testing.utils
-   :members: