Paper
Beyond Query Linearity
Nonlinear query projections in transformer attention. Parameter-matched residual MLP queries beat the linear baseline and a model with 12.5% more non-embedding parameters. Headed to an ICLR 2026 workshop.
- PyTorch
- Transformers
- arXiv 2603.13381