ColossalAI/examples/tutorial/hybrid_parallel/requirements.txt