ColossalAI/colossalai/nn/layer/colossalai_layer
..
__init__.py
_utils.py
dropout.py
embedding.py
linear.py
normalization.py