Latest, Oct 9, 2026
A custom attention kernel for Gemma 4 on the RTX PRO 6000
FlashAttention has no kernel for Gemma 4’s largest attention heads on the RTX PRO 6000. Ours cuts global attention from 1,770 ms to 464 ms per 20k-token request, against a hardware floor of 294 ms.
vLLM default
1,770 ms
Our kernel
464 ms
floor 294 ms
- Before
- 1,770 ms
- After
- 464 ms
- Hardware floor
- 294 ms
- Left to the floor
- 170 ms