About this paper
FLAMIA accelerates Transformer attention on NVIDIA Blackwell GPUs via MXFP8 and mixed-precision kernels that reduce quantization overhead, prevent causal leakage, and support variable-length packed documents.
Attention remains a substantial computational cost in Transformer training, particularly at long context lengths. NVIDIA Blackwell GPUs provide high-throughput low-precision matrix multiplication, but exploiting it in attention requires addressing quantization overhead, causal information leakage through shared scales, and numerical inconsistencies between the forward and backward passes. We present FLAMIA, a family of attention kernels for Blackwell with MXFP8 and mixed NVFP4/FP8 training paths. Our design combines token-block scaling that reuses quantized operands across matrix orientations, targeted safeguards against scale-induced leakage, and consistent forward and backward…
attention kernelsMXFP8NVFP4FP8token-block scalingcausal information leakageforward and backward passesfused quantizationdocument-local scalingpacked layouts
- Upvotes
- 8
- Authors
- 3
- Published
- Nov 2, 2026
Summary and abstract excerpt via Hugging Face Papers ↗ · arXiv.
Trending history
FLAMIA: FLAsh MIcroscaling Attention for Fast and Causally Safe Training on Blackwell was first seen on Hugging Face · Daily papers on Nov 3, 2026. Since then it has appeared in 10 checks on 2 different days. Its best position was #11 of 30, reached on Nov 4, 05:28 UTC. It dropped off the list after Nov 4, 05:28 UTC.