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