ColossalAI/examples/tutorial/sequence_parallel/loss_func
..
__init__.py
bert_loss.py
cross_entropy.py
utils.py