Files
ColossalAI/examples/tutorial/sequence_parallel/loss_func