FlyWheel Accelerates TPU Softmax Attention by 4.14x

FlyWheel optimizes TPU attention kernels by reformulating softmax and triangular inversion to minimize vector work and improve matrix operand reuse. Implementation on TPU v6e achieves 81% of peak BF16 throughput, providing a 1.56x prefill throughput boost for Qwen3.8-27B in vLLM.