colossalai.nn.parallel.layers.embedding ======================================= .. automodule:: colossalai.nn.parallel.layers.embedding :members: