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