PyPTX Surpasses FA4 on B300

A_K_Nain · x · 2026-07-16

PyPTX Flash Attention is compared to FlashAttention 4 on B300, and the cited post claims it's faster on all metrics: BF16 up to 7.1%, FP8 up to 6.9%, using strict paired timing.

It also claims PyPTX has better numerical precision than FA4. The poster says "it's ramping up," hinting at further optimization potential.

Original post →

More from Infra

Infra channel →