3

Broken Symmetry in BF16 Attention: Why FlashAttention Gradients Blow Up Late in Training
BF16 is now standard in large-scale pretraining, including in fused attention kernels such as FlashAttention, and these kernels are …
Junlin Chen, Daize Dong, Huanwei Di, Haolong Jia, Jiawei Wu, Haotian Xie, Mingkai Zheng, Yang Li, Leshang Chen, Huishu Wang, Eric P. Xing, Hongyi Wang
PDF Cite arXiv
Diamond Agent: Agentic Control of Federated HPC Resources as a Service
Efficiently aggregating and orchestrating computing power across heterogeneous clusters for HPC workflows faces four practical …
Haotian Xie, Junlin Chen, Mingkai Zheng, Yifan Zhu, Minu Mathew, Max Burnette, Yadu Babuji, Volodymyr Kindratenko, Shivaram Venkataraman, Kyle Chard, Ian Foster, Zhao Zhang
PDF Cite arXiv
PHOENIX: Resilient LLM Training with Hot-Swapping via Zero-Overhead Checkpoint
State-of-the-art large language model (LLM) training takes tens of thousands of graphics processing units (GPUs) for months and …
Haotian Xie, Junlin Chen, Mingkai Zheng, Lishan Yang, Zhao Zhang
PDF Cite arXiv
SCAPE: Accurate and Efficient LLM Training with Extreme Sparse Communication
Communication increasingly dominates the cost of Large Language Model (LLM) pre-training, especially under data-parallel and sharded …
Mingkai Zheng, Junlin Chen, Haotian Xie, Zhao Zhang
PDF Cite arXiv