[doc] update rst (#470)

* update rst

* remove empty rst

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

+colossalai.amp.naive\_amp.grad\_scaler.base\_grad\_scaler
+=========================================================
+
+.. automodule:: colossalai.amp.naive_amp.grad_scaler.base_grad_scaler
+   :members:

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

+colossalai.amp.naive\_amp.grad\_scaler.constant\_grad\_scaler
+=============================================================
+
+.. automodule:: colossalai.amp.naive_amp.grad_scaler.constant_grad_scaler
+   :members:

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

+colossalai.amp.naive\_amp.grad\_scaler.dynamic\_grad\_scaler
+============================================================
+
+.. automodule:: colossalai.amp.naive_amp.grad_scaler.dynamic_grad_scaler
+   :members:

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

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

docs/colossalai/colossalai.context.rst

@@ -15,5 +15,6 @@ colossalai.context
    :maxdepth: 2
 
    colossalai.context.config
+   colossalai.context.moe_context
    colossalai.context.parallel_context
    colossalai.context.parallel_mode

docs/colossalai/colossalai.core.rst

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

docs/colossalai/colossalai.engine.gradient_handler.rst

@@ -3,3 +3,9 @@ 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:

docs/colossalai/colossalai.engine.ophooks.rst

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

docs/colossalai/colossalai.engine.ophooks.zero_hook.rst

+colossalai.engine.ophooks.zero\_hook
+====================================
+
+.. automodule:: colossalai.engine.ophooks.zero_hook
+   :members:

docs/colossalai/colossalai.engine.paramhooks.rst

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

docs/colossalai/colossalai.engine.rst

@@ -8,4 +8,6 @@ colossalai.engine
    :maxdepth: 2
 
    colossalai.engine.gradient_handler
+   colossalai.engine.ophooks
+   colossalai.engine.paramhooks
    colossalai.engine.schedule

docs/colossalai/colossalai.zero.loss_scaler.rst → docs/colossalai/colossalai.global_variables.rst

-colossalai.zero.loss\_scaler
+colossalai.global\_variables
 ============================
 
-.. automodule:: colossalai.zero.loss_scaler
+.. automodule:: colossalai.global_variables
    :members:

docs/colossalai/colossalai.nn.layer.moe.experts.rst

+colossalai.nn.layer.moe.experts
+===============================
+
+.. automodule:: colossalai.nn.layer.moe.experts
+   :members:

docs/colossalai/colossalai.nn.layer.moe.rst

@@ -8,4 +8,6 @@ colossalai.nn.layer.moe
 .. toctree::
    :maxdepth: 2
 
+   colossalai.nn.layer.moe.experts
    colossalai.nn.layer.moe.layers
+   colossalai.nn.layer.moe.utils

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

+colossalai.nn.layer.moe.utils
+=============================
+
+.. automodule:: colossalai.nn.layer.moe.utils
+   :members:

docs/colossalai/colossalai.nn.loss.loss_1d.rst

+colossalai.nn.loss.loss\_1d
+===========================
+
+.. automodule:: colossalai.nn.loss.loss_1d
+   :members: