Field noteNov 3, 20266 min read1 source

FLAMIA: FLAsh MIcroscaling Attention for Fast and Causally Safe Training on Blackwell

In one minute

FLAMIA is a family of attention kernels for NVIDIA Blackwell GPUs that combines MXFP8 and mixed NVFP4/FP8 training paths. Its authors report 1.10–1.40× acceleration on B200 and 1.06–1.42× on B300 for combined forward-and-backward attention, including quantization, while preserving stable Llama-style pretraining in their test. For teams training on Blackwell, the work shows why low-precision gains require numerical and kernel design together.

What changed: FLAMIA is a new Blackwell attention-kernel paper, reporting MXFP8 speedups and training-stability results not covered by the archive.

What happened

Fedor Velikonivtsev, Mikhail Koltakov, and Denis Kuznedelev presented FLAMIA in an arXiv paper submitted on November 2, 2026. The work targets the attention operation during Transformer training, particularly at long context lengths. According to the abstract, NVIDIA Blackwell GPUs provide high-throughput low-precision matrix multiplication. Applying that throughput to attention is less straightforward: the authors identify quantization overhead, causal information leakage through shared scales, and numerical mismatches between forward and backward passes as obstacles.

FLAMIA is described as a family of Blackwell-specific attention kernels with two training paths: MXFP8 and a mixed NVFP4/FP8 route. The design uses token-block scaling so that quantized operands can be reused across matrix orientations. This matters because repeatedly producing low-precision representations can consume enough work to erode a theoretical matrix-multiplication gain. The authors also describe fused quantization and architecture-specific kernel optimization as ways to reduce that overhead.

The causal-safety issue is specific to quantized attention with shared scaling values. A scale that is calculated over a region containing future tokens can allow information about those tokens to influence earlier positions, despite a causal mask. The authors say FLAMIA includes targeted safeguards against this scale-induced leakage. They also aim for consistent forward and backward computations, rather than treating a fast forward kernel as sufficient evidence for training use.

The kernels are designed for packed layouts containing documents of different lengths. Document-local scaling and workload-aware scheduling are intended to support those variable-length documents directly, without data repacking. This is relevant to the reported evaluation because the results concern causal packed-document workloads, rather than a single fixed-shape benchmark. The supplied material does not specify how each scheduling choice performs independently; it presents the components as part of the FLAMIA design.

For the MXFP8 path, the authors report acceleration of the combined forward-and-backward attention call, including quantization, of 1.10–1.40× on NVIDIA B200 and 1.06–1.42× on B300. Each number is relative to the fastest evaluated baseline for that workload, not to one universal baseline. The project description names FlashAttention-4 in BF16, cuDNN, LP-FA4, and FlashInfer NVFP4 among the implementations tested. That framing is important: the range describes differing workload and baseline outcomes, rather than a promised fixed improvement over any particular library.

The paper also reports validation on pretraining Llama-style models. The supplied project description specifies a Llama-like model with 1 billion parameters trained on 100 billion tokens. It says training remained stable and final metrics were close to BF16. The evidence is therefore about a research implementation and a defined training experiment. It does not establish a package release, license, installation path, hosted service, or a supported interface that another team can adopt immediately.

What the archive adds

This is a new event within an ongoing attention-optimization thread. FeatherAttn: Fast attention on RDNA3/3.5 reported an AMD attention-backward kernel with claimed speedups on RDNA3/3.5. Nunchux AI's VC-Attention speeds up MiniMax-H3 attention without retraining covered a training-free optimization. FLAMIA instead focuses on Blackwell training, low-precision paths, and the complete forward/backward attention call including quantization.

Trend signal

None of the supplied patterns precisely captures accelerator-specific attention-kernel co-design. FLAMIA weakly relates to agent economics moves to completed work: it reinforces the broader idea that a useful efficiency claim must be measured at the level of an operational workload, not a favorable isolated primitive. But its evidence is still training-kernel speed, not cost per verified completed task.

Why now

The paper was submitted on November 2, 2026 and is framed around Blackwell's low-precision matrix-multiplication capability. Its timing follows from an implementation question created by that hardware capability: whether FP8 and FP4 throughput can translate into faster causal attention training once quantization, backward computation, packed documents, and training numerics are included.

Connections to other events

The archive's FeatherAttn note concerned attention-backward performance on AMD RDNA3/3.5, while VC-Attention reported attention acceleration without retraining on MiniMax-H3. Together, these reports show attention optimization being pursued across different hardware and operating constraints. FLAMIA's distinct contribution is its claim that causal safety and backward-pass consistency must be addressed alongside low-bit performance in a Blackwell training kernel.

Why it matters

For a team training models on B200 or B300 systems, attention remains a potentially material share of the training path, especially as contexts get longer. FLAMIA is useful less as a generic claim that FP8 or FP4 is faster than as a concrete checklist for evaluation: count quantization time, evaluate forward and backward together, use the actual packed-document layout, and test whether causal behavior and optimization stability remain intact.

The reported ranges justify benchmarking the approach against the fastest baseline relevant to a team's workload. They do not justify assuming the upper end of the range will apply to every sequence length or data layout. Teams on other accelerators, or teams using managed training where custom kernels are unavailable, have far less immediate use for the result.

What happens next

  • Blackwell training teams will test low-precision attention as an end-to-end kernel path rather than an isolated matrix-multiplication substitution. (6 months, 60%). Watch for: benchmarks reporting quantization and backward-pass time together.
  • Further work will separate causal-safety and quality results by workload and precision path. (12 months, 55%). Watch for: expanded evaluations beyond the reported Llama-style experiment.

Limits and uncertainty

The supplied public source is an arXiv abstract plus a project description. It does not establish peer review, independent reproduction, production adoption, or long-term maintenance. The speedups are reported as ranges over evaluated workloads and against the fastest evaluated baseline for each workload, so they are not a single fixed comparison. The material gives no absolute runtimes, energy measurements, dollar savings, or results for other accelerator architectures.

It also does not provide a license, code distribution, API, compatibility matrix, or deployment instructions. “Close to BF16” is not quantified in the supplied material, and the described training test does not by itself prove comparable behavior for different model scales, datasets, context lengths, or optimizer settings.

Skill to build now

If your team operates Blackwell training infrastructure, build a benchmark harness that measures full attention-step time rather than only matrix-multiplication throughput. Include quantization, forward and backward work, representative sequence-length distributions, and the packed-document layout used in production. Record numerical checks for causal masking and training stability beside performance figures. That discipline will make it possible to determine whether a low-precision kernel creates a real training gain for your workload.

Reader takeaway

FLAMIA reports meaningful Blackwell attention gains by treating low precision, quantization overhead, causal safety, packed-document scheduling, and training numerics as one system-design problem. The evidence is promising for B200/B300 training teams, but it remains workload-dependent research rather than a portable performance guarantee or an immediately installable developer tool.

Open question

Will FLAMIA's reported gains persist at larger model scales and across the sequence-length and document-packing distributions used in production training?

Evidence / sources

  1. 1arxiv.org/abs/2611.03344
All Field Notes

Next reads

Read next

Open the graph
  1. 01topicInferenceExplore the inference topic hub.
  2. 02topicModelsExplore the models topic hub.
  3. 03topicComputeExplore the compute topic hub.
  4. 04archiveField NotesOpen the latest editorial analysis.
  5. 05sourcesSourcesSee the original sources this archive cites.
Who read this page?Machine requests, hidden until opened

Loading the privacy-safe route aggregate...

Open the JSON contract