[doc] update rst and docstring (#1351)

* update rst

* add zero docstr

* fix docstr

* remove fx.tracer.meta_patch

* fix docstr

* fix docstr

* update fx rst

* fix fx docstr

* remove useless rst

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

@@ -8,5 +8,4 @@ colossalai.nn.layer.wrapper
 .. toctree::
    :maxdepth: 2
 
-   colossalai.nn.layer.wrapper.lambda_wrapper
    colossalai.nn.layer.wrapper.pipeline_wrapper

docs/colossalai/colossalai.nn.optimizer.rst

@@ -16,3 +16,4 @@ colossalai.nn.optimizer
    colossalai.nn.optimizer.hybrid_adam
    colossalai.nn.optimizer.lamb
    colossalai.nn.optimizer.lars
+   colossalai.nn.optimizer.utils

docs/colossalai/colossalai.nn.optimizer.utils.rst

+colossalai.nn.optimizer.utils
+=============================
+
+.. automodule:: colossalai.nn.optimizer.utils
+   :members:

docs/colossalai/colossalai.nn.parallel.data_parallel.rst

+colossalai.nn.parallel.data\_parallel
+=====================================
+
+.. automodule:: colossalai.nn.parallel.data_parallel
+   :members:

docs/colossalai/colossalai.nn.parallel.layers.colo_module.rst

+colossalai.nn.parallel.layers.colo\_module
+==========================================
+
+.. automodule:: colossalai.nn.parallel.layers.colo_module
+   :members:

docs/colossalai/colossalai.utils.profiler.mem_profiler.rst → docs/colossalai/colossalai.nn.parallel.layers.embedding.rst

-colossalai.utils.profiler.mem\_profiler
+colossalai.nn.parallel.layers.embedding
 =======================================
 
-.. automodule:: colossalai.utils.profiler.mem_profiler
+.. automodule:: colossalai.nn.parallel.layers.embedding
    :members:

docs/colossalai/colossalai.nn.parallel.layers.linear.rst

+colossalai.nn.parallel.layers.linear
+====================================
+
+.. automodule:: colossalai.nn.parallel.layers.linear
+   :members:

docs/colossalai/colossalai.nn.parallel.layers.module_utils.rst

+colossalai.nn.parallel.layers.module\_utils
+===========================================
+
+.. automodule:: colossalai.nn.parallel.layers.module_utils
+   :members:

docs/colossalai/colossalai.nn.parallel.layers.rst

+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

docs/colossalai/colossalai.nn.parallel.reducer.rst

+colossalai.nn.parallel.reducer
+==============================
+
+.. automodule:: colossalai.nn.parallel.reducer
+   :members:

docs/colossalai/colossalai.nn.parallel.rst

+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

docs/colossalai/colossalai.nn.rst

@@ -7,12 +7,13 @@ colossalai.nn
 .. toctree::
    :maxdepth: 2
 
+   colossalai.nn.graph
    colossalai.nn.layer
    colossalai.nn.loss
    colossalai.nn.lr_scheduler
    colossalai.nn.metric
-   colossalai.nn.model
    colossalai.nn.optimizer
+   colossalai.nn.parallel
 
 
 .. toctree::

docs/colossalai/colossalai.pipeline.layer_sepc.rst

+colossalai.pipeline.layer\_sepc
+===============================
+
+.. automodule:: colossalai.pipeline.layer_sepc
+   :members:

docs/colossalai/colossalai.pipeline.pipelinable.rst

+colossalai.pipeline.pipelinable
+===============================
+
+.. automodule:: colossalai.pipeline.pipelinable
+   :members:

docs/colossalai/colossalai.pipeline.rst

+colossalai.pipeline
+===================
+
+.. automodule:: colossalai.pipeline
+   :members:
+
+
+.. toctree::
+   :maxdepth: 2
+
+   colossalai.pipeline.layer_sepc
+   colossalai.pipeline.pipelinable
+   colossalai.pipeline.utils

docs/colossalai/colossalai.pipeline.utils.rst

+colossalai.pipeline.utils
+=========================
+
+.. automodule:: colossalai.pipeline.utils
+   :members:

docs/colossalai/colossalai.rst

 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
@@ -22,4 +30,7 @@ colossalai
 .. toctree::
    :maxdepth: 2
 
+   colossalai.constants
+   colossalai.core
+   colossalai.global_variables
    colossalai.initialize

docs/colossalai/colossalai.tensor.colo_parameter.rst

+colossalai.tensor.colo\_parameter
+=================================
+
+.. automodule:: colossalai.tensor.colo_parameter
+   :members:

docs/colossalai/colossalai.tensor.colo_tensor.rst

+colossalai.tensor.colo\_tensor
+==============================
+
+.. automodule:: colossalai.tensor.colo_tensor
+   :members:

docs/colossalai/colossalai.tensor.compute_spec.rst

+colossalai.tensor.compute\_spec
+===============================
+
+.. automodule:: colossalai.tensor.compute_spec
+   :members: