Files
happy-llm/docs/chapter5/code/ddp_pretrain.py