Test-Time Regression

The blocked matrix formulation of linear attention mechanisms, multi-step online gradient descent at inference time, and chunk-wise parallelism.

A unifying framework for linear attention mechanisms as test-time regression and how to parallelize training and inference.