ColossalAI/model_zoo
..
bert
gpt
mlp_mixer
moe
vit
__init__.py
helper.py