Latest, Oct 9, 2026
A custom attention kernel for Gemma 4 on the RTX PRO 6000
No version of FlashAttention runs Gemma 4’s attention over the whole prompt on the RTX PRO 6000, so we wrote our own kernel.
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