ColossalAI/examples/tutorial/large_batch_optimizer/requirements.txt