ColossalAI/examples/tutorial/auto_parallel/requirements.txt