update examples and sphnix docs for the new api (#63)

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

 colossalai.nn.layer.parallel\_sequence
 ======================================
 
-.. automodule:: colossalai.nn.layer.parallel_sequence
-   :members:
-
-
 .. toctree::
    :maxdepth: 2
 
    colossalai.nn.layer.parallel_sequence.layers
+
+
+.. automodule:: colossalai.nn.layer.parallel_sequence
+   :members:

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

-colossalai.nn.layer.parallel\_vision\_transformer.layers
-========================================================
-
-.. automodule:: colossalai.nn.layer.parallel_vision_transformer.layers
-   :members:

docs/colossalai/colossalai.nn.layer.parallel_vision_transformer.rst

-colossalai.nn.layer.parallel\_vision\_transformer
-=================================================
-
-.. automodule:: colossalai.nn.layer.parallel_vision_transformer
-   :members:
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.nn.layer.parallel_vision_transformer.layers

docs/colossalai/colossalai.nn.layer.rst

 colossalai.nn.layer
 ===================
 
-.. automodule:: colossalai.nn.layer
-   :members:
-
 .. toctree::
    :maxdepth: 2
 
@@ -12,13 +9,10 @@ colossalai.nn.layer
    colossalai.nn.layer.parallel_2p5d
    colossalai.nn.layer.parallel_3d
    colossalai.nn.layer.parallel_sequence
-   colossalai.nn.layer.parallel_vision_transformer
-   colossalai.nn.layer.vanilla_resnet
-   colossalai.nn.layer.vanilla_vision_transformer
+   colossalai.nn.layer.non_parallel_layers
    colossalai.nn.layer.wrapper
+   colossalai.nn.layer.base_layer
 
 
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.nn.layer.base_layer
+.. automodule:: colossalai.nn.layer
+   :members:

docs/colossalai/colossalai.nn.layer.vanilla_resnet.basic_block.rst

-colossalai.nn.layer.vanilla\_resnet.basic\_block
-================================================
-
-.. automodule:: colossalai.nn.layer.vanilla_resnet.basic_block
-   :members:

docs/colossalai/colossalai.nn.layer.vanilla_resnet.bottleneck.rst

-colossalai.nn.layer.vanilla\_resnet.bottleneck
-==============================================
-
-.. automodule:: colossalai.nn.layer.vanilla_resnet.bottleneck
-   :members:

docs/colossalai/colossalai.nn.layer.vanilla_resnet.conv.rst

-colossalai.nn.layer.vanilla\_resnet.conv
-========================================
-
-.. automodule:: colossalai.nn.layer.vanilla_resnet.conv
-   :members:

docs/colossalai/colossalai.nn.layer.vanilla_resnet.reslayer.rst

-colossalai.nn.layer.vanilla\_resnet.reslayer
-============================================
-
-.. automodule:: colossalai.nn.layer.vanilla_resnet.reslayer
-   :members:

docs/colossalai/colossalai.nn.layer.vanilla_resnet.rst

-colossalai.nn.layer.vanilla\_resnet
-===================================
-
-.. automodule:: colossalai.nn.layer.vanilla_resnet
-   :members:
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.nn.layer.vanilla_resnet.basic_block
-   colossalai.nn.layer.vanilla_resnet.bottleneck
-   colossalai.nn.layer.vanilla_resnet.conv
-   colossalai.nn.layer.vanilla_resnet.reslayer

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

-colossalai.nn.layer.vanilla\_vision\_transformer.layers
-=======================================================
-
-.. automodule:: colossalai.nn.layer.vanilla_vision_transformer.layers
-   :members:

docs/colossalai/colossalai.nn.layer.vanilla_vision_transformer.rst

-colossalai.nn.layer.vanilla\_vision\_transformer
-================================================
-
-.. automodule:: colossalai.nn.layer.vanilla_vision_transformer
-   :members:
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.nn.layer.vanilla_vision_transformer.layers

docs/colossalai/colossalai.nn.loss.base_loss.rst

-colossalai.nn.loss.base\_loss
-=============================
-
-.. automodule:: colossalai.nn.loss.base_loss
-   :members:

docs/colossalai/colossalai.nn.loss.cross_entropy_1d.rst

-colossalai.nn.loss.cross\_entropy\_1d
-=====================================
-
-.. automodule:: colossalai.nn.loss.cross_entropy_1d
-   :members:

docs/colossalai/colossalai.nn.loss.rst

 colossalai.nn.loss
 ==================
 
-.. automodule:: colossalai.nn.loss
-   :members:
-
-
 .. toctree::
    :maxdepth: 2
 
-   colossalai.nn.loss.base_loss
-   colossalai.nn.loss.cross_entropy_1d
    colossalai.nn.loss.cross_entropy_2d
    colossalai.nn.loss.cross_entropy_2p5d
    colossalai.nn.loss.cross_entropy_3d
+
+
+.. automodule:: colossalai.nn.loss
+   :members:

docs/colossalai/colossalai.nn.lr_scheduler.rst

 colossalai.nn.lr\_scheduler
 ===========================
 
-.. automodule:: colossalai.nn.lr_scheduler
-   :members:
-
-
 .. toctree::
    :maxdepth: 2
 
@@ -15,3 +11,7 @@ colossalai.nn.lr\_scheduler
    colossalai.nn.lr_scheduler.onecycle
    colossalai.nn.lr_scheduler.poly
    colossalai.nn.lr_scheduler.torch
+
+   
+.. automodule:: colossalai.nn.lr_scheduler
+   :members:

docs/colossalai/colossalai.nn.model.base_model.rst → docs/colossalai/colossalai.nn.model.model_from_config.rst

-colossalai.nn.model.base\_model
+colossalai.nn.model.model\_from\_config
 ===============================
 
-.. automodule:: colossalai.nn.model.base_model
+.. automodule:: colossalai.nn.model.model_from_config
    :members:

docs/colossalai/colossalai.nn.model.rst

 colossalai.nn.model
 ===================
 
-.. automodule:: colossalai.nn.model
-   :members:
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.nn.model.vanilla_resnet
-   colossalai.nn.model.vision_transformer
-
-
 .. toctree::
    :maxdepth: 2
 
-   colossalai.nn.model.base_model
+   colossalai.nn.model.model_from_config

docs/colossalai/colossalai.nn.model.vanilla_resnet.resnet.rst

-colossalai.nn.model.vanilla\_resnet.resnet
-==========================================
-
-.. automodule:: colossalai.nn.model.vanilla_resnet.resnet
-   :members:

docs/colossalai/colossalai.nn.model.vanilla_resnet.rst

-colossalai.nn.model.vanilla\_resnet
-===================================
-
-.. automodule:: colossalai.nn.model.vanilla_resnet
-   :members:
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.nn.model.vanilla_resnet.resnet

docs/colossalai/colossalai.nn.model.vision_transformer.rst

-colossalai.nn.model.vision\_transformer
-=======================================
-
-.. automodule:: colossalai.nn.model.vision_transformer
-   :members:
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.nn.model.vision_transformer.vision_transformer