Field noteOct 31, 20265 min read1 source

feather-attn: Fast attention on RDNA3/3.5

In one minute

FeatherAttn is an MIT-licensed open-source attention backward kernel for AMD RDNA3/3.5 GPUs. Its author reports up to a 4x speedup against AITER’s tuned FlashAttention Triton backward kernel on a Strix Halo system, offering PyTorch training users a drop-in path that retains AITER for forward execution and unsupported cases.

What changed: FeatherAttn adds an AMD RDNA3/3.5 attention-backward kernel, reported up to 4× faster than AITER’s FlashAttention Triton kernel on Strix Halo.

What happened

Developer woct0rdho has published FeatherAttn, a public GitHub package described as fast attention for AMD’s RDNA3 and RDNA3.5 GPU architectures. The package’s current scope is deliberately narrow: it provides a fast attention backward kernel. Backward attention is part of the training computation, rather than the forward-only path typically associated with serving an already trained model. That focus makes FeatherAttn relevant to teams fine-tuning or training workloads on the supported AMD hardware, not a general replacement for every attention implementation.

The repository reports tests on Strix Halo. Against AITER’s FlashAttention Triton backward kernel with tuned configurations, the author says FeatherAttn can be four times as fast at head dimension 128 and 1.3 times as fast at head dimension 64. These are implementation-specific comparisons, not a broad benchmark of AMD GPUs, model families, batch sizes, sequence lengths, or training workloads. The README also reports approximately 40% faster Anima LoRA training, but does not establish how that result transfers to other LoRA jobs or systems.

FeatherAttn includes a trainable attention function intended as a drop-in replacement for flash_attn_func. The replacement combines AITER’s forward kernel with FeatherAttn’s backward kernel. Where FeatherAttn’s backward kernel does not support a feature, the package falls back to AITER’s backward kernel. This design matters because it frames the release as an incremental substitution inside an existing PyTorch training script rather than a demand to rewrite a training stack around a new interface.

The package supports torch.compile, according to the repository, so it can be wired into a compiled PyTorch workflow. It is released under the MIT license. The author explicitly says that the current FeatherAttn backward kernel does not use FP8 quantization. The project name refers to an idea of fast FP8 matrix multiplication on RDNA3/3.5 hardware without native FP8 support, but FP8 quantization for attention forward and backward kernels remains ongoing work rather than a shipped FeatherAttn capability.

The comparison baseline is important. FeatherAttn does not claim to replace AITER end to end: AITER remains its forward-kernel provider and its fallback for unsupported backward features. The practical proposition is therefore a targeted optimization of the backward pass for a particular AMD architecture family, with compatibility behavior designed to preserve execution when the specialized path cannot be used.

What the archive adds

This is a new fact on a related GPU-kernel story. OpenBMB’s ForgeKernel release concerned a different GPU kernel library and an optimization agent; FeatherAttn instead ships one specialized AMD attention backward implementation. Both point to kernel-level work becoming a visible lever in AI-system performance, but they do not establish the same performance result or deployment path.

Trend signal

No supplied pattern genuinely fits closely enough. FeatherAttn is a low-level implementation release, while the available patterns address agent-product architecture, interoperability, security, organizational workflows, and verification rather than accelerator-kernel optimization.

Why it matters

For builders training or fine-tuning on supported AMD consumer hardware, attention backward performance can be a practical constraint rather than an abstract systems metric. FeatherAttn presents a low-integration-cost experiment: replace a familiar attention function, retain AITER for the forward pass, and allow fallback where the new kernel lacks support. That is a more usable adoption shape than a standalone benchmark kernel with no compatibility story.

The release also illustrates how to read vendor-adjacent performance claims. A reported 4x result at one head dimension can be meaningful for a matching workload, but it is not a guarantee of 4x faster training. The useful unit of evaluation is end-to-end job time on the team’s model, data shape, sequence length, memory conditions, and ROCm environment. The author’s reported LoRA result is closer to that unit, but remains a single reported workload.

What happens next

  • FeatherAttn will receive broader workload or hardware validation beyond Strix Halo. (3 months, 55%). Watch for: published results on additional RDNA3/3.5 systems and training configurations.
  • FP8 attention work will remain an extension rather than a currently usable FeatherAttn feature. (6 months, 60%). Watch for: a release that explicitly adds FP8 quantization to forward or backward attention.

Limits and uncertainty

The public repository supplies author-reported results, not an independent benchmark suite. It identifies Strix Halo as the tested system but does not establish performance on all RDNA3/3.5 GPUs, nor does it provide enough information here to generalize across model architectures, sequence lengths, batch sizes, head counts, or ROCm versions. The stated speedups compare against AITER’s tuned FlashAttention Triton backward kernel, not every alternative attention stack. Feature coverage is incomplete by design, since unsupported backward features fall back to AITER. The source also does not establish numerical accuracy, stability, installation friction, or production readiness for a particular training environment.

Skill to build now

Treat the package as a measured substitution experiment. In a representative PyTorch fine-tuning job on supported AMD hardware, record baseline wall-clock time and memory behavior with the existing attention path, then repeat with FeatherAttn’s flash_attn_func replacement and confirm that unsupported features take the documented fallback route. Compare whole-job results, not only kernel timing, and retain a rollback path to AITER.

Reader takeaway

FeatherAttn is not a universal attention stack; it is a focused AMD RDNA3/3.5 backward-pass optimization with a compatibility-oriented integration path. Its reported gains make it worth testing for matching PyTorch training workloads, while its single-system, author-reported evidence means teams should validate their own end-to-end speedup before treating it as infrastructure.

Open question

Which training shapes and RDNA3/3.5 devices preserve FeatherAttn’s claimed gains once fallback behavior, numerical validation, and full-job runtime are measured?

Evidence / sources

  1. 1github.com/woct0rdho/feather-attn
All Field Notes

Next reads

Read next

Open the graph
  1. 01archiveField NotesOpen the latest editorial analysis.
  2. 02source ledgerSource LedgerInspect the public source evidence graph.
Who read this page?Machine requests, hidden until opened

Loading the privacy-safe route aggregate...

Open the JSON contract