Engineering ·
BF16 rounding may destabilize FlashAttention gradients
A September 29, 2026 report says BF16 rounding in FlashAttention may break a softmax-gradient symmetry and cause gradients to explode near the end of training. The authors propose GProj as a correction.