colossalai.nn.layer.parallel\_2d ================================ .. automodule:: colossalai.nn.layer.parallel_2d :members: