Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving
Not on the map. Why?
Abstract (arXiv)
Feedforward computation, such as evaluating a neural network or sampling from an autoregressive model, is ubiquitous in machine learning. The sequential nature of feedforward computation, however, requires a strict order of execution and cannot be easily accelerated with parallel computing. To enable parallelization, we frame the task of feedforward computation as solving a system of nonlinear equations. We then propose to find the solution using a Jacobi or Gauss-Seidel fixed-point iteration method, as well as hybrid methods of both. Crucially, Jacobi updates operate independently on each equation and can be executed in parallel. Our method is guaranteed to give exactly the same values as the original feedforward computation with a reduced (or equal) number of parallelizable iterations, and hence reduced time given sufficient parallel computing power. Experimentally, we demonstrate the effectiveness of our approach in accelerating (i) backpropagation of RNNs, (ii) evaluation of DenseNets, and (iii) autoregressive sampling of MADE and PixelCNN++, with speedup factors between 2.1 and 26 under various settings.
Key facts. Cited by 52 papers in the map (data through 14 Oct 2026). First posted on arXiv 10 Feb 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 3; 2022: 2; 2023: 6; 2024: 12; 2025: 15; 2026: 14.
52 papers.
Sort:
By area:
- 2023Accelerating Large Language Model Decoding with Speculative Samplingcited by 1,006
- 2024Break the Sequential Dependency of LLM Inference Using Lookahead Decodingcited by 298
- 2022Continuous diffusion for categorical datacited by 214
- 2021Autoregressive Diffusion Modelscited by 210
- 2024LLM Inference Unveiled: Survey and Roofline Model Insightscited by 152
- 2022Concrete Score Matching: Generalized Score Matching for Discrete Datacited by 148
- 2023Accelerating Transformer Inference for Translation via Parallel Decodingcited by 124
- 2023Parallel Sampling of Diffusion Modelscited by 112
- 2025SimpleAR: Pushing the Frontier of Autoregressive Visual Generation through Pretraining, SFT, and RLcited by 86
- 2024CLLMs: Consistency Large Language Modelscited by 72
- 2024Accelerating Auto-regressive Text-to-Image Generation with Training-free Speculative Jacobi Decodingcited by 66
- 2024Parallelized Autoregressive Visual Generationcited by 60
38 more
- 2023Parallelizing non-linear sequential models over the sequence lengthcited by 45
- 2024Towards Scalable and Stable Parallelization of Nonlinear RNNscited by 38
- 2025Lumina-mGPT 2.0: Stand-Alone AutoRegressive Image Modelingcited by 37
- 2024Accelerating Parallel Sampling of Diffusion Modelscited by 30
- 2023DreamPropeller: Supercharge Text-to-3D Generation with Parallel Samplingcited by 20
- 2024Distilled Decoding 1: One-step Sampling of Image Auto-regressive Models with Flow Matchingcited by 18
- 2025Grouped Speculative Decoding for Autoregressive Image Generationcited by 17
- 2025ParaRNN: Unlocking Parallel Training of Nonlinear RNNs for Large Language Modelscited by 15
- 2025Fast and Accurate Causal Parallel Decoding using Jacobi Forcingcited by 13
- 2023DeepPCR: Parallelizing Sequential Operations in Neural Networkscited by 9
- 2024Inference Optimization of Foundation Models on AI Acceleratorscited by 9
- 2025STARFlow-V: End-to-End Video Generative Modeling with Normalizing Flowscited by 9
- 2024Parallel Sampling via Countingcited by 8
- 2024Faster Language Models with Better Multi-Token Prediction Using Tensor Decompositioncited by 8
- 2026xPress: Parallel Refinement for Diffusion Drafters in Speculative Decodingcited by 8
- 2025Predictability Enables Parallelization of Nonlinear State Space Modelscited by 7
- 2025Speculative Jacobi-Denoising Decoding for Accelerating Autoregressive Text-to-image Generationcited by 7
- 2024S3D: A Simple and Cost-Effective Self-Speculative Decoding Scheme for Low-Memory GPUscited by 6
- 2025A Unifying Framework for Parallelizing Sequential Models with Linear Dynamical Systemscited by 4
- 2025Speculative Coupled Decoding for Training-Free Lossless Acceleration of Autoregressive Visual Generationcited by 4
- 2026The Coupling Within: Flow Matching via Distilled Normalizing Flowscited by 3
- 2025PCM : Picard Consistency Model for Fast Parallel Sampling of Diffusion Modelscited by 2
- 2025Accelerate TarFlow Sampling with GS-Jacobi Iterationcited by 2
- 2025SJD++: Improved Speculative Jacobi Decoding for Training-free Acceleration of Discrete Auto-regressive Text-to-Image Generationcited by 2
- 2026Parallel Jacobi Decoding for Fast Autoregressive Image Generationcited by 2
- 2026Accelerating Speculative Diffusions via Block Verificationcited by 2
- 2021PixelPyramids: Exact Inference Models from Lossless Image Pyramidscited by 1
- 2021ELF: Exact-Lipschitz Based Universal Density Approximator Flowcited by 1
- 2025Accelerating Inference of Discrete Autoregressive Normalizing Flows by Selective Jacobi Decodingcited by 1
- 2026SJD-PV: Speculative Jacobi Decoding with Phrase Verification for Autoregressive Image Generationcited by 1
- 2026SJD-VP: Speculative Jacobi Decoding with Verification Prediction for Autoregressive Image Generationcited by 1
- 2026SNLP: Layer-Parallel Inference via Structured Newton Correctionscited by 1
- 2025Acceleration Multiple Heads Decoding for LLM via Dynamic Tree Attentioncited by 0
- 2026The Context-Ready Transformercited by 0
- 2026CSD: Content-aware Speculative Decoding for Efficient Image Generationcited by 0
- 2026Layer-Parallel Inference Reduces Encrypted Nonlinear Depth in Transformerscited by 0
- 2026SJD-SV: Speculative Jacobi Decoding with Semantics Verification for Autoregressive Image Generationcited by 0
- 2026GradLev: Token-Parallel Test-Time Training Via Costate Predictioncited by 0
50 most cited of 52 All 52 (JSON)
- 2026JetSpec++: Foresight Parallel Drafting with Self Refinements14 Oct 2026
- 2026Switching Linear Attention30 Sep 2026
- 2026GradLev: Token-Parallel Test-Time Training Via Costate Prediction28 Sep 2026
- 2026SJD-SV: Speculative Jacobi Decoding with Semantics Verification for Autoregressive Image Generation4 Sep 2026
- 2026xPress: Parallel Refinement for Diffusion Drafters in Speculative Decoding3 Aug 2026
- 2026Layer-Parallel Inference Reduces Encrypted Nonlinear Depth in Transformers6 Jul 2026
- 2026CSD: Content-aware Speculative Decoding for Efficient Image Generation26 Jun 2026
- 2026The Context-Ready Transformer25 Jun 2026
- 2026Accelerating Speculative Diffusions via Block Verification11 Jun 2026
- 2026Parallel Jacobi Decoding for Fast Autoregressive Image Generation4 Jun 2026
- 2026SNLP: Layer-Parallel Inference via Structured Newton Corrections18 May 2026
- 2026SJD-VP: Speculative Jacobi Decoding with Verification Prediction for Autoregressive Image Generation28 Mar 2026
38 more
- 2026The Coupling Within: Flow Matching via Distilled Normalizing Flows9 Mar 2026
- 2026SJD-PV: Speculative Jacobi Decoding with Phrase Verification for Autoregressive Image Generation2 Mar 2026
- 2025Fast and Accurate Causal Parallel Decoding using Jacobi Forcing16 Dec 2025
- 2025SJD++: Improved Speculative Jacobi Decoding for Training-free Acceleration of Discrete Auto-regressive Text-to-Image Generation8 Dec 2025
- 2025STARFlow-V: End-to-End Video Generative Modeling with Normalizing Flows25 Nov 2025
- 2025Speculative Coupled Decoding for Training-Free Lossless Acceleration of Autoregressive Visual Generation28 Oct 2025
- 2025ParaRNN: Unlocking Parallel Training of Nonlinear RNNs for Large Language Models24 Oct 2025
- 2025Speculative Jacobi-Denoising Decoding for Accelerating Autoregressive Text-to-image Generation10 Oct 2025
- 2025A Unifying Framework for Parallelizing Sequential Models with Linear Dynamical Systems26 Sep 2025
- 2025Predictability Enables Parallelization of Nonlinear State Space Models22 Aug 2025
- 2025Grouped Speculative Decoding for Autoregressive Image Generation11 Aug 2025
- 2025Lumina-mGPT 2.0: Stand-Alone AutoRegressive Image Modeling23 Jul 2025
- 2025Accelerating Inference of Discrete Autoregressive Normalizing Flows by Selective Jacobi Decoding30 May 2025
- 2025Accelerate TarFlow Sampling with GS-Jacobi Iteration19 May 2025
- 2025SimpleAR: Pushing the Frontier of Autoregressive Visual Generation through Pretraining, SFT, and RL15 Apr 2025
- 2025PCM : Picard Consistency Model for Fast Parallel Sampling of Diffusion Models25 Mar 2025
- 2025Acceleration Multiple Heads Decoding for LLM via Dynamic Tree Attention9 Feb 2025
- 2024Distilled Decoding 1: One-step Sampling of Image Auto-regressive Models with Flow Matching22 Dec 2024
- 2024Parallelized Autoregressive Visual Generation19 Dec 2024
- 2024Faster Language Models with Better Multi-Token Prediction Using Tensor Decomposition23 Oct 2024
- 2024Accelerating Auto-regressive Text-to-Image Generation with Training-free Speculative Jacobi Decoding2 Oct 2024
- 2024Parallel Sampling via Counting18 Aug 2024
- 2024Towards Scalable and Stable Parallelization of Nonlinear RNNs26 Jul 2024
- 2024Inference Optimization of Foundation Models on AI Accelerators12 Jul 2024
- 2024S3D: A Simple and Cost-Effective Self-Speculative Decoding Scheme for Low-Memory GPUs30 May 2024
- 2024CLLMs: Consistency Large Language Models28 Feb 2024
- 2024LLM Inference Unveiled: Survey and Roofline Model Insights26 Feb 2024
- 2024Accelerating Parallel Sampling of Diffusion Models15 Feb 2024
- 2024Break the Sequential Dependency of LLM Inference Using Lookahead Decoding3 Feb 2024
- 2023DreamPropeller: Supercharge Text-to-3D Generation with Parallel Sampling28 Nov 2023
- 2023DeepPCR: Parallelizing Sequential Operations in Neural Networks28 Sep 2023
- 2023Parallelizing non-linear sequential models over the sequence length21 Sep 2023
- 2023Parallel Sampling of Diffusion Models25 May 2023
- 2023Accelerating Transformer Inference for Translation via Parallel Decoding17 May 2023
- 2023Accelerating Large Language Model Decoding with Speculative Sampling2 Feb 2023
- 2022Continuous diffusion for categorical data28 Nov 2022
- 2022Concrete Score Matching: Generalized Score Matching for Discrete Data2 Nov 2022
- 2021ELF: Exact-Lipschitz Based Universal Density Approximator Flow13 Dec 2021
50 newest of 52 All 52 (JSON)
- 2023Accelerating Large Language Model Decoding with Speculative Samplingcited by 1,006
- 2024Break the Sequential Dependency of LLM Inference Using Lookahead Decodingcited by 298
- 2024LLM Inference Unveiled: Survey and Roofline Model Insightscited by 152
- 2023Accelerating Transformer Inference for Translation via Parallel Decodingcited by 124
- 2024CLLMs: Consistency Large Language Modelscited by 72
- 2024Accelerating Auto-regressive Text-to-Image Generation with Training-free Speculative Jacobi Decodingcited by 66
- 2025Fast and Accurate Causal Parallel Decoding using Jacobi Forcingcited by 13
- 2026xPress: Parallel Refinement for Diffusion Drafters in Speculative Decodingcited by 8
- 2024S3D: A Simple and Cost-Effective Self-Speculative Decoding Scheme for Low-Memory GPUscited by 6
- 2025SJD++: Improved Speculative Jacobi Decoding for Training-free Acceleration of Discrete Auto-regressive Text-to-Image Generationcited by 2
- 2026Accelerating Speculative Diffusions via Block Verificationcited by 2
- 2026CSD: Content-aware Speculative Decoding for Efficient Image Generationcited by 0
12 most cited of 13 All 13 (JSON)
- 2026JetSpec++: Foresight Parallel Drafting with Self Refinements14 Oct 2026
- 2026xPress: Parallel Refinement for Diffusion Drafters in Speculative Decoding3 Aug 2026
- 2026CSD: Content-aware Speculative Decoding for Efficient Image Generation26 Jun 2026
- 2026Accelerating Speculative Diffusions via Block Verification11 Jun 2026
- 2025Fast and Accurate Causal Parallel Decoding using Jacobi Forcing16 Dec 2025
- 2025SJD++: Improved Speculative Jacobi Decoding for Training-free Acceleration of Discrete Auto-regressive Text-to-Image Generation8 Dec 2025
- 2024Accelerating Auto-regressive Text-to-Image Generation with Training-free Speculative Jacobi Decoding2 Oct 2024
- 2024S3D: A Simple and Cost-Effective Self-Speculative Decoding Scheme for Low-Memory GPUs30 May 2024
- 2024CLLMs: Consistency Large Language Models28 Feb 2024
- 2024LLM Inference Unveiled: Survey and Roofline Model Insights26 Feb 2024
- 2024Break the Sequential Dependency of LLM Inference Using Lookahead Decoding3 Feb 2024
- 2023Accelerating Transformer Inference for Translation via Parallel Decoding17 May 2023
12 newest of 13 All 13 (JSON)
- 2022Continuous diffusion for categorical datacited by 214
- 2021Autoregressive Diffusion Modelscited by 210
- 2023Parallel Sampling of Diffusion Modelscited by 112
- 2024Accelerating Parallel Sampling of Diffusion Modelscited by 30
- 2023DreamPropeller: Supercharge Text-to-3D Generation with Parallel Samplingcited by 20
- 2024Distilled Decoding 1: One-step Sampling of Image Auto-regressive Models with Flow Matchingcited by 18
- 2025Grouped Speculative Decoding for Autoregressive Image Generationcited by 17
- 2023DeepPCR: Parallelizing Sequential Operations in Neural Networkscited by 9
- 2026The Coupling Within: Flow Matching via Distilled Normalizing Flowscited by 3
- 2025PCM : Picard Consistency Model for Fast Parallel Sampling of Diffusion Modelscited by 2
- 2026The Coupling Within: Flow Matching via Distilled Normalizing Flows9 Mar 2026
- 2025Grouped Speculative Decoding for Autoregressive Image Generation11 Aug 2025
- 2025PCM : Picard Consistency Model for Fast Parallel Sampling of Diffusion Models25 Mar 2025
- 2024Distilled Decoding 1: One-step Sampling of Image Auto-regressive Models with Flow Matching22 Dec 2024
- 2024Accelerating Parallel Sampling of Diffusion Models15 Feb 2024
- 2023DreamPropeller: Supercharge Text-to-3D Generation with Parallel Sampling28 Nov 2023
- 2023DeepPCR: Parallelizing Sequential Operations in Neural Networks28 Sep 2023
- 2023Parallel Sampling of Diffusion Models25 May 2023
- 2022Continuous diffusion for categorical data28 Nov 2022
- 2021Autoregressive Diffusion Models5 Oct 2021
- 2022Concrete Score Matching: Generalized Score Matching for Discrete Datacited by 148
- 2025SimpleAR: Pushing the Frontier of Autoregressive Visual Generation through Pretraining, SFT, and RLcited by 86
- 2024Parallelized Autoregressive Visual Generationcited by 60
- 2023Parallelizing non-linear sequential models over the sequence lengthcited by 45
- 2024Towards Scalable and Stable Parallelization of Nonlinear RNNscited by 38
- 2025Lumina-mGPT 2.0: Stand-Alone AutoRegressive Image Modelingcited by 37
- 2025ParaRNN: Unlocking Parallel Training of Nonlinear RNNs for Large Language Modelscited by 15
- 2024Inference Optimization of Foundation Models on AI Acceleratorscited by 9
- 2025STARFlow-V: End-to-End Video Generative Modeling with Normalizing Flowscited by 9
- 2024Parallel Sampling via Countingcited by 8
- 2024Faster Language Models with Better Multi-Token Prediction Using Tensor Decompositioncited by 8
- 2025Predictability Enables Parallelization of Nonlinear State Space Modelscited by 7
12 most cited of 29 All 29 (JSON)
- 2026Switching Linear Attention30 Sep 2026
- 2026GradLev: Token-Parallel Test-Time Training Via Costate Prediction28 Sep 2026
- 2026SJD-SV: Speculative Jacobi Decoding with Semantics Verification for Autoregressive Image Generation4 Sep 2026
- 2026Layer-Parallel Inference Reduces Encrypted Nonlinear Depth in Transformers6 Jul 2026
- 2026The Context-Ready Transformer25 Jun 2026
- 2026Parallel Jacobi Decoding for Fast Autoregressive Image Generation4 Jun 2026
- 2026SNLP: Layer-Parallel Inference via Structured Newton Corrections18 May 2026
- 2026SJD-VP: Speculative Jacobi Decoding with Verification Prediction for Autoregressive Image Generation28 Mar 2026
- 2026SJD-PV: Speculative Jacobi Decoding with Phrase Verification for Autoregressive Image Generation2 Mar 2026
- 2025STARFlow-V: End-to-End Video Generative Modeling with Normalizing Flows25 Nov 2025
- 2025Speculative Coupled Decoding for Training-Free Lossless Acceleration of Autoregressive Visual Generation28 Oct 2025
- 2025ParaRNN: Unlocking Parallel Training of Nonlinear RNNs for Large Language Models24 Oct 2025
12 newest of 29 All 29 (JSON)