[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.gemini.chunk_mgr.rst

+colossalai.gemini.chunk\_mgr
+============================
+
+.. automodule:: colossalai.gemini.chunk_mgr
+   :members:

docs/colossalai/colossalai.gemini.gemini_context.rst

+colossalai.gemini.gemini\_context
+=================================
+
+.. automodule:: colossalai.gemini.gemini_context
+   :members:

docs/colossalai/colossalai.gemini.gemini_mgr.rst

+colossalai.gemini.gemini\_mgr
+=============================
+
+.. automodule:: colossalai.gemini.gemini_mgr
+   :members:

docs/colossalai/colossalai.utils.memory_tracer.async_memtracer.rst → docs/colossalai/colossalai.gemini.memory_tracer.memory_monitor.rst

-colossalai.utils.memory\_tracer.async\_memtracer
+colossalai.gemini.memory\_tracer.memory\_monitor
 ================================================
 
-.. automodule:: colossalai.gemini.memory_tracer.async_memtracer
+.. automodule:: colossalai.gemini.memory_tracer.memory_monitor
    :members:

docs/colossalai/colossalai.utils.memory_tracer.memstats_collector.rst → docs/colossalai/colossalai.gemini.memory_tracer.memstats_collector.rst

-colossalai.utils.memory\_tracer.memstats\_collector
-===================================================
+colossalai.gemini.memory\_tracer.memstats\_collector
+====================================================
 
 .. automodule:: colossalai.gemini.memory_tracer.memstats_collector
    :members:

docs/colossalai/colossalai.utils.memory_tracer.model_data_memtracer.rst → docs/colossalai/colossalai.gemini.memory_tracer.model_data_memtracer.rst

-colossalai.utils.memory\_tracer.model\_data\_memtracer
-======================================================
+colossalai.gemini.memory\_tracer.model\_data\_memtracer
+=======================================================
 
 .. automodule:: colossalai.gemini.memory_tracer.model_data_memtracer
    :members:

docs/colossalai/colossalai.utils.memory_tracer.rst → docs/colossalai/colossalai.gemini.memory_tracer.rst

-colossalai.utils.memory\_tracer
-===============================
+colossalai.gemini.memory\_tracer
+================================
 
 .. automodule:: colossalai.gemini.memory_tracer
    :members:
@@ -8,6 +8,6 @@ colossalai.utils.memory\_tracer
 .. toctree::
    :maxdepth: 2
 
-   colossalai.gemini.memory_tracer.async_memtracer
+   colossalai.gemini.memory_tracer.memory_monitor
    colossalai.gemini.memory_tracer.memstats_collector
    colossalai.gemini.memory_tracer.model_data_memtracer

docs/colossalai/colossalai.engine.ophooks.rst → docs/colossalai/colossalai.gemini.ophooks.rst

@@ -8,4 +8,4 @@ colossalai.gemini.ophooks
 .. toctree::
    :maxdepth: 2
 
-   colossalai.gemini.ophooks.zero_hook
+   colossalai.gemini.ophooks.utils

docs/colossalai/colossalai.gemini.ophooks.utils.rst

+colossalai.gemini.ophooks.utils
+===============================
+
+.. automodule:: colossalai.gemini.ophooks.utils
+   :members:

docs/colossalai/colossalai.gemini.paramhooks.rst

+colossalai.gemini.paramhooks
+============================
+
+.. automodule:: colossalai.gemini.paramhooks
+   :members:

docs/colossalai/colossalai.gemini.placement_policy.rst

+colossalai.gemini.placement\_policy
+===================================
+
+.. automodule:: colossalai.gemini.placement_policy
+   :members:

docs/colossalai/colossalai.gemini.rst

+colossalai.gemini
+=================
+
+.. automodule:: colossalai.gemini
+   :members:
+
+.. toctree::
+   :maxdepth: 2
+
+   colossalai.gemini.memory_tracer
+   colossalai.gemini.ophooks
+   colossalai.gemini.paramhooks
+
+
+.. toctree::
+   :maxdepth: 2
+
+   colossalai.gemini.chunk
+   colossalai.gemini.chunk_mgr
+   colossalai.gemini.gemini_context
+   colossalai.gemini.gemini_mgr
+   colossalai.gemini.placement_policy
+   colossalai.gemini.stateful_tensor
+   colossalai.gemini.stateful_tensor_container
+   colossalai.gemini.stateful_tensor_mgr
+   colossalai.gemini.tensor_placement_policy
+   colossalai.gemini.tensor_utils

docs/colossalai/colossalai.gemini.stateful_tensor.rst

+colossalai.gemini.stateful\_tensor
+==================================
+
+.. automodule:: colossalai.gemini.stateful_tensor
+   :members:

docs/colossalai/colossalai.gemini.stateful_tensor_container.rst

+colossalai.gemini.stateful\_tensor\_container
+=============================================
+
+.. automodule:: colossalai.gemini.stateful_tensor_container
+   :members:

docs/colossalai/colossalai.nn.model.model_from_config.rst → docs/colossalai/colossalai.gemini.stateful_tensor_mgr.rst

-colossalai.nn.model.model\_from\_config
+colossalai.gemini.stateful\_tensor\_mgr
 =======================================
 
-.. automodule:: colossalai.nn.model.model_from_config
+.. automodule:: colossalai.gemini.stateful_tensor_mgr
    :members:

docs/colossalai/colossalai.nn.layer.wrapper.lambda_wrapper.rst → docs/colossalai/colossalai.gemini.tensor_placement_policy.rst

-colossalai.nn.layer.wrapper.lambda\_wrapper
+colossalai.gemini.tensor\_placement\_policy
 ===========================================
 
-.. automodule:: colossalai.nn.layer.wrapper.lambda_wrapper
+.. automodule:: colossalai.gemini.tensor_placement_policy
    :members:

docs/colossalai/colossalai.gemini.tensor_utils.rst

+colossalai.gemini.tensor\_utils
+===============================
+
+.. automodule:: colossalai.gemini.tensor_utils
+   :members:

docs/colossalai/colossalai.nn.graph.graph_node.rst

+colossalai.nn.graph.graph\_node
+===============================
+
+.. automodule:: colossalai.nn.graph.graph_node
+   :members:

docs/colossalai/colossalai.nn.model.rst → docs/colossalai/colossalai.nn.graph.rst

-colossalai.nn.model
+colossalai.nn.graph
 ===================
 
-.. automodule:: colossalai.nn.model
+.. automodule:: colossalai.nn.graph
    :members:
 
 
 .. toctree::
    :maxdepth: 2
 
-   colossalai.nn.model.model_from_config
+   colossalai.nn.graph.graph_node
+   colossalai.nn.graph.utils

docs/colossalai/colossalai.nn.graph.utils.rst

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