ColossalAI/examples/tutorial/sequence_parallel/model/layers
..
__init__.py
bert_layer.py
dropout.py
embedding.py
head.py
init_method.py
linear.py
mlp.py
pooler.py
preprocess.py