Systems and Algorithms for Convolutional Multi-Hybrid Language Models at Scale
Not on the map. Why?
Abstract (arXiv)
We introduce convolutional multi-hybrid architectures, with a design grounded on two simple observations. First, operators in hybrid models can be tailored to token manipulation tasks such as in-context recall, multi-token recall, and compression, with input-dependent convolutions and attention offering complementary performance. Second, co-designing convolution operators and hardware-aware algorithms enables efficiency gains in regimes where previous alternative architectures struggle to surpass Transformers. At the 40 billion parameter scale, we train end-to-end 1.2 to 2.9 times faster than optimized Transformers, and 1.1 to 1.4 times faster than previous generation hybrids. On H100 GPUs and model width 4096, individual operators in the proposed multi-hybrid StripedHyena 2 architecture achieve two-fold throughput improvement over linear attention and state-space models. Multi-hybrids excel at sequence modeling over byte-tokenized data, as demonstrated by the Evo 2 line of models. We discuss the foundations that enable these results, including architecture design, overlap-add blocked kernels for tensor cores, and dedicated all-to-all and point-to-point context parallelism strategies.
Key facts. Cited by 12 papers in the map (data through 7 Oct 2026). Builds on 47 of its 55 references that are in the map. First posted on arXiv 25 Feb 2025.
Builds on
47 of 55 references.
References · 47
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
- 2020Longformer: The Long-Document Transformercited by 3,362
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
- 2020GLU Variants Improve Transformercited by 1,724
- 2019Generating Long Sequences with Sparse Transformerscited by 1,523
- 2019Root Mean Square Layer Normalizationcited by 1,425
- 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,311
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,283
35 more
- 2019ZeRO: Memory Optimizations Toward Training Trillion Parameter Modelscited by 1,278
- 2024DeepSeek-V2: A Strong, Economical, and Efficient Mixture-of-Experts Language Modelcited by 1,093
- 2020On Layer Normalization in the Transformer Architecturecited by 906
- 2023RWKV: Reinventing RNNs for the Transformer Eracited by 814
- 2023Extending Context Window of Large Language Models via Positional Interpolationcited by 721
- 2019Fast Transformer Decoding: One Write-Head is All You Needcited by 700
- 2023PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallelcited by 641
- 2021CoAtNet: Marrying Convolution and Attention for All Data Sizescited by 566
- 2024FlashAttention-3: Fast and Accurate Attention with Asynchrony and Low-precisioncited by 507
- 2022Hungry Hungry Hippos: Towards Language Modeling with State Space Modelscited by 495
- 2023Ring Attention with Blockwise Transformers for Near-Infinite Contextcited by 448
- 2023Resurrecting Recurrent Neural Networks for Long Sequencescited by 397
- 2024Jamba: A Hybrid Transformer-Mamba Language Modelcited by 390
- 2023Hyena Hierarchy: Towards Larger Convolutional Language Modelscited by 388
- 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 373
- 2022Diagonal State Spaces are as Effective as Structured State Spacescited by 350
- 2024xLSTM: Extended Long Short-Term Memorycited by 320
- 2023Effective Long-Context Scaling of Foundation Modelscited by 301
- 2018An Empirical Model of Large-Batch Trainingcited by 268
- 2014cuDNN: Efficient Primitives for Deep Learningcited by 240
- 2023DeepSpeed Ulysses: System Optimizations for Enabling Training of Extreme Long Sequence Transformer Modelscited by 239
- 2021Scaling Local Self-Attention for Parameter Efficient Visual Backbonescited by 220
- 2024Repeat After Me: Transformers are Better than State Space Models at Copyingcited by 208
- 2023Zoology: Measuring and Improving Recall in Efficient Language Modelscited by 188
- 2024Zamba: A Compact 7B SSM Hybrid Modelcited by 118
- 2021CKConv: Continuous Kernel Convolution For Sequential Datacited by 117
- 2024In-Context Language Learning: Architectures and Algorithmscited by 86
- 2024Mechanistic Design and Scaling of Hybrid Architecturescited by 57
- 2023Striped Attention: Faster Ring Attention for Causal Transformerscited by 52
- 2024Jamba-1.5: Hybrid Transformer-Mamba Models at Scalecited by 48
- 2023FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Corescited by 32
- 2017Parallel Multi Channel Convolution using General Matrix Multiplicationcited by 30
- 2023Laughing Hyena Distillery: Extracting Compact Recurrences From Convolutionscited by 24
- 2024Training Ultra Long Context Language Model with Fully Pipelined Distributed Transformercited by 6
- 2021tcFFT: Accelerating Half-Precision FFT through Tensor Corescited by 5
Built on by
Citing papers per year: 2025: 7; 2026: 5.
12 papers.
- 2025Exploring Diffusion Transformer Designs via Graftingcited by 12
- 2025Panda: Self-distillation of Reusable Sensor-level Representations for High Energy Physicscited by 2
- 2025Best Practices for Biorisk Evaluations on Open-Weight Bio-Foundation Modelscited by 1
- 2025Sliding Window Recurrences for Sequence Modelscited by 1
- 2026MambaFormer: Token-Level Guided Routing Mixture-of-Experts for Accurate and Efficient Clinical Assistancecited by 1
- 2025Evaluating DNA function understanding in genomic language models using evolutionarily implausible sequencescited by 0
- 2025ShishuLM : Achieving Optimal and Efficient Parameterization with Low Attention Transformer Modelscited by 0
- 2025GraphCliff: Short-Long Range Gating for Modeling Critical Activity Changes Caused by Subtle Molecular Differencescited by 0
- 2026Flux Attention: Context-Aware Hybrid Attention for Efficient LLMs Inferencecited by 0
- 2026In Search of Lost DNA Sequence Pretrainingcited by 0
- 2026Yeti: A compact protein structure tokenizer for reconstruction and multi-modal generationcited by 0
- 2026Native Multi-Dimensional Subquadratic Operators via Input Dependent Long Convolutionscited by 0
- 2026Native Multi-Dimensional Subquadratic Operators via Input Dependent Long Convolutions1 Jul 2026
- 2026Yeti: A compact protein structure tokenizer for reconstruction and multi-modal generation11 May 2026
- 2026In Search of Lost DNA Sequence Pretraining17 Apr 2026
- 2026Flux Attention: Context-Aware Hybrid Attention for Efficient LLMs Inference8 Apr 2026
- 2026MambaFormer: Token-Level Guided Routing Mixture-of-Experts for Accurate and Efficient Clinical Assistance3 Jan 2026
- 2025Sliding Window Recurrences for Sequence Models15 Dec 2025
- 2025Panda: Self-distillation of Reusable Sensor-level Representations for High Energy Physics1 Dec 2025
- 2025GraphCliff: Short-Long Range Gating for Modeling Critical Activity Changes Caused by Subtle Molecular Differences5 Nov 2025
- 2025Best Practices for Biorisk Evaluations on Open-Weight Bio-Foundation Models31 Oct 2025
- 2025ShishuLM : Achieving Optimal and Efficient Parameterization with Low Attention Transformer Models13 Oct 2025
- 2025Evaluating DNA function understanding in genomic language models using evolutionarily implausible sequences12 Jun 2025
- 2025Exploring Diffusion Transformer Designs via Grafting5 Jun 2025
Similar, not cited
Close in wording, no citation either way
- 2025Hybrid Architectures for Language Models: Systematic Analysis and Design Insights
- 2025Understanding and Enhancing Mamba-Transformer Hybrids for Memory Recall and Language Modeling
- 2026Expressivity-Efficiency Tradeoffs for Hybrid Sequence Models
- 2025Multiscale Byte Language Models -- A Hierarchical Architecture for Causal Million-Length Sequence Modeling
- 2026HybridKV: Hybrid KV Cache Compression for Efficient Multimodal Large Language Model Inference
- 2026Long-Context Modeling via GSS-Transformer Hybrid Architecture with Learnable Mixing
- 2026DASH: Fast Differentiable Architecture Search for Hybrid Attention in Minutes on a Single GPU
- 2026HybridGen: Efficient LLM Generative Inference via CPU-GPU Hybrid Computing