Files
ColossalAI/examples/tutorial/hybrid_parallel/train.py