A Unifying Framework for Parallelizing Sequential Models with Linear Dynamical Systems
Not on the map. Why?
Abstract (arXiv)
Harnessing parallelism in seemingly sequential models is a central challenge for modern machine learning. Several approaches have been proposed for evaluating sequential processes in parallel using iterative fixed-point methods, like Newton, Picard, and Jacobi iterations. In this work, we show that these methods can be understood within a common framework based on linear dynamical systems (LDSs), where different iteration schemes arise naturally as approximate linearizations of a nonlinear recursion. Moreover, we theoretically analyze the rates of convergence of these methods, and we verify the predictions of this theory with several case studies. This unifying framework highlights shared principles behind these techniques and clarifies when particular fixed-point methods are most likely to be effective. By bridging diverse algorithms through the language of LDSs, the framework provides a clearer theoretical foundation for parallelizing sequential models and points toward new opportunities for efficient and scalable computation.
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 43 of its 92 references that are in the map. First posted on arXiv 26 Sep 2025.
Builds on
43 of 92 references.
References · 43
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,101
- 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,057
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
- 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,336
31 more
- 2022Simplified State Space Layers for Sequence Modelingcited by 788
- 2019Deep Equilibrium Modelscited by 615
- 2021Combining Recurrent, Convolutional, and Continuous-time Models with Linear State-Space Layerscited by 608
- 2023Gated Linear Attention Transformers with Hardware-Efficient Trainingcited by 514
- 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 415
- 2023Resurrecting Recurrent Neural Networks for Long Sequencescited by 408
- 2024xLSTM: Extended Long Short-Term Memorycited by 327
- 2024The Illusion of State in State-Space Modelscited by 187
- 2021Equinox: neural networks in JAX via callable PyTrees and filtered transformationscited by 148
- 2017Parallelizing Linear Recurrent Neural Nets Over Sequence Lengthcited by 127
- 2016Composing graphical models with neural networks for structured representations and fast inferencecited by 113
- 2023Parallel Sampling of Diffusion Modelscited by 112
- 2024Unlocking State-Tracking in Linear RNNs Through Negative Eigenvaluescited by 101
- 2020Accelerating Feedforward Computation via Parallel Nonlinear Equation Solvingcited by 52
- 2023Parallelizing non-linear sequential models over the sequence lengthcited by 45
- 2023Entity Tracking in Language Modelscited by 44
- 2024Towards Scalable and Stable Parallelization of Nonlinear RNNscited by 38
- 2019MintNet: Building Invertible Neural Networks with Masked Convolutionscited by 35
- 2019Temporal Parallelization of Bayesian Smootherscited by 33
- 2024Accelerating Parallel Sampling of Diffusion Modelscited by 30
- 2025Implicit Language Models are RNNs: Balancing Parallelization and Expressivitycited by 22
- 2023Revisiting Structured Variational Autoencoderscited by 11
- 2025Parallelization of Non-linear State-Space Models: Scaling Up Liquid-Resistance Liquid-Capacitance Networks for Efficient Sequence Modelingcited by 10
- 2023DeepPCR: Parallelizing Sequential Operations in Neural Networkscited by 9
- 2024Self-Refining Diffusion Samplers: Enabling Parallelization via Parareal Iterationscited by 7
- 2025Predictability Enables Parallelization of Nonlinear State Space Modelscited by 7
- 2022Parallel square-root statistical linear regression for inference in nonlinear state space modelscited by 6
- 2021Temporal Parallelization of Inference in Hidden Markov Modelscited by 5
- 2025SING: SDE Inference via Natural Gradientscited by 5
- 2017Parallel Complexity of Forward and Backward Propagationcited by 3
- 2025Parallel computations for Metropolis Markov chains with Picard mapscited by 1
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026Why Are Linear RNNs More Parallelizable?cited by 8
- 2026SNLP: Layer-Parallel Inference via Structured Newton Correctionscited by 1
- 2026Riccati State Space Models: Non-iterative Parallelization for Nonlinear Sequence Modelingcited by 0
- 2026Switching Linear Attentioncited by 0
- 2026Switching Linear Attention30 Sep 2026
- 2026Riccati State Space Models: Non-iterative Parallelization for Nonlinear Sequence Modeling28 Sep 2026
- 2026SNLP: Layer-Parallel Inference via Structured Newton Corrections18 May 2026
- 2026Why Are Linear RNNs More Parallelizable?4 Mar 2026