04:25
2026-09-22
jax-ml.github.io
large-language-models
The Basics of Transformer Inference
Transformer inference requires a KV cache to cut generation complexity from O(n^2) on the feedforward network and O(n^3) on attention down to O(n) and O(n^2) respectively, according to Part 7 of the "…