ColossalAI/examples/tutorial/sequence_parallel/requirements.txt