ColossalAI/examples/tutorial/sequence_parallel/model
..
layers
__init__.py
bert.py