REVIEW 4 major objections 5 minor 38 references
Diffusion language models do not need attention: a bidirectional Mamba-2 denoiser preserves masked-diffusion quality while making decoding cost linear in sequence length.
Reviewed by Pith at T0; open to challenge. T0 means a machine referee read the full paper against a public rubric. the ladder, T0–T4 →
T0 review · deepseek-v4-flash
2026-08-03 21:14 UTC pith:WYUW3LRA
load-bearing objection First bidirectional Mamba-2 denoiser for masked diffusion LMs in text, with a real throughput win — but the abstract oversells a different paper and the quality comparison is confounded by parameter mismatch. the 4 major comments →
DiffuMamba: High-Throughput Diffusion LMs with Mamba Backbone
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
At its core, the paper claims that absorbing-state masked discrete diffusion is agnostic to the sequence mixer: the denoiser's attention layers can be replaced by bidirectional Mamba-2 layers and the probabilistic structure of the diffusion process, the masked corruption, the reweighted masked cross-entropy objective, and the factorized unmasking sampler, remains unchanged. Empirically, at 1.3B parameters, the hybrid denoiser DiffuApriel-H+MLP reduces validation perplexity from 25.01 to 22.89 under one compute budget and from 22.72 to 20.17 under another, and it beats the attention baseline on all seven zero-shot perplexity benchmarks and on average downstream reasoning accuracy by roughly 3
What carries the argument
The central object is the bidirectional Mamba-2 mixer: each diffusion block runs two input-dependent selective state-space recurrences, one forward and one backward along the sequence, adds their outputs, and applies timestep-conditioned adaptive layer normalization. This entirely replaces quadratic self-attention in the denoiser, giving O(B L d) per-block cost. In the hybrid variant, one attention layer is inserted every K=5 Mamba blocks, yielding effective complexity O(B L d + B L^2 d / K), so the linear term dominates for moderate lengths. The same masked-diffusion training loss and the same factorized unmasking sampler are retained; the architectural change is confined to the denoiser's
Load-bearing premise
The paper's quality conclusion assumes its controlled comparison isolates the architecture choice: the Mamba-based models are compared to a Transformer baseline with fewer parameters at every scale (about 15-30% fewer), so if extra capacity, not the state-space design, explains the better perplexity and accuracy, the central claim that state-space denoisers are stronger would not be established.
What would settle it
Run the paper's exact 1.3B training setup with a Transformer denoiser matched to DiffuApriel-H's parameter count or FLOPs, keeping data, tokenizer, noise schedule, and 128 decoding steps identical; if the Transformer then matches or beats the hybrid on validation perplexity and downstream average, the claim that state-space denoisers are the better backbone is refuted. Separately, throughput can be settled directly: at 128K context on one H100, DiffuApriel tokens/sec should stay approximately flat while DiffuTran decays roughly as 1/L.
If this is right
- At 1.3B scale, DiffuApriel+MLP measures 4.4x end-to-end tokens/sec over the attention baseline and DiffuApriel-H+MLP 2.6x, with the gap widening as sequence length grows.
- While Transformer throughput decays roughly as 1/L beyond a critical length, the Mamba model's throughput saturates at a memory-bandwidth floor, avoiding the long-context degradation seen with KV-cache block diffusion.
- The hybrid Mamba-attention denoiser beats the attention-only baseline on validation perplexity and all seven zero-shot benchmarks at 1.3B, and the pure Mamba variant ranks second on most.
- Interleaving infrequent attention (every five blocks) with state-space layers gives the best quality-efficiency tradeoff in the paper, extending the autoregressive hybrid finding to diffusion.
- The mixer swap does not disturb the absorbing-state diffusion framework, so existing DLM acceleration techniques can compose with the new backbone.
Where Pith is reading between the lines
- The paper fixes attention frequency at every 5 blocks and never varies it; if the hybrid margin over pure Mamba grows with scale, the optimal attention density may depend on model size, which is untested.
- An equal-parameter or equal-FLOP head-to-head run would separate architecture from capacity; the paper's own configurations leave this undetermined.
- The throughput plateau below peak memory bandwidth suggests a direct serving measurement beyond 64K tokens would test whether the near-linear scaling holds in practice, especially for the hybrid.
- Because block masking is described as orthogonal and the hybrid already pairs local and global mixers, combining block diffusion with Mamba denoisers seems a natural next experiment that the paper does not run.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper (body title 'DiffuApriel', arXiv metadata title 'DiffuMamba') proposes replacing the Transformer denoiser of masked diffusion language models with a bidirectional Mamba-2 backbone, plus a hybrid variant with interleaved attention. It claims that these models preserve the probabilistic structure of masked discrete diffusion while enabling linear-time inference, and reports quality and throughput comparisons against a Transformer-based DiffuTran baseline at 240M, 0.5B, and 1.3B scales, with up to 4.4×/2.6× throughput gains (the metadata abstract claims 8.2×/4.3×). The paper also contains complexity analysis and wall-clock throughput measurements on an H100.
Significance. If the results hold, this would be the first demonstration that SSM denoisers are viable for discrete diffusion language modeling, a non-obvious and useful result: it decouples diffusion LM quality from quadratic attention and points to a concrete path for long-context diffusion inference. The hybrid exploration and the attempt at controlled scaling comparisons are also valuable. The throughput mechanism (linear vs. quadratic per-step cost) is sound in principle and does not depend on the confounded quality comparison. However, the central architectural-quality claim — that replacing attention with Mamba 'does not compromise quality' — is currently not established because the compared models differ substantially in parameter count.
major comments (4)
- [§4, Table 1] The controlled-comparison premise is violated. Table 1 lists DiffuTran-1.3B at 1.3B params but DiffuApriel-1.6B at 1.6B (+23%) and DiffuApriel-H-1.5B at 1.5B (+15%); at 0.5B the gaps are +29% (0.67B) and +17% (0.61B) versus 0.52B; at 240M the gaps are +21% and +8%. Yet §4 states 'the only difference lies in the internal mixer architecture' and claims an 'architecture-isolated evaluation.' Halving the MLP expansion ratio does not equalize total parameters. Consequently Tables 2–4 and the statement 'DiffuApriel-H+MLP surpasses DiffuTran across all datasets' may express a scale advantage rather than an SSM advantage. The throughput claims are unaffected, but the quality claim needs either matched-parameter models, same-compute retraining at equal parameter count, or a clear re-framing that the comparison is at budgeted-compute, not architecture-isolated, settings.
- [§4.2, Figure 2] The decoding protocol for the KV-cache baseline is inconsistent with the rest of the comparison. The text states that DiffuTran+KV caching uses 'a block size of 32 with 2 denoising steps per block,' while the caption says 'constant 128 decoding steps' for all methods. If the KV baseline uses a different total number of denoising steps or a different step schedule, the throughput comparison is not controlled. Please report the exact decoding step count, schedule, and block size for every curve, or remove the KV-cache comparison from the headline throughput claims.
- [Abstract (arXiv metadata) vs. body] The arXiv abstract claims 'up to 8.2x and 4.3x higher inference throughput' and states that 'cache-efficient block diffusion with Mamba mixers emerges as the only strategy that scales linearly...' — neither appears in the body. The body reports 4.4× and 2.6× throughput and explicitly defers block diffusion to future work (§4.2, §5). The version deposited under arXiv:2511.15927 therefore misrepresents the paper's actual results. The abstract and body must be aligned, and the unsupported block-diffusion claim must be removed or supported with experiments.
- [§3.2, Eqs. (4)–(6)] The formal definition of the bidirectional Mamba block is not faithful to Mamba-2 and is not reproducible as written. Equations (4)–(5) describe fixed learnable kernels A_f, B_f applied as 1D convolutions/scan recurrences, while Mamba-2's selective SSM is input-dependent (B, C are projections of the input) and computed via structured state-space duality, not a simple additive fusion of forward/backward scans as in Eq. (6). If the implementation actually uses the Mamba-2 kernel, the equations should be replaced with the exact block specification or a pseudocode reference; otherwise readers cannot tell whether the experiments use standard Mamba-2 or a simplified recurrent layer.
minor comments (5)
- [Title/header] The body title is 'DiffuApriel' while the arXiv metadata title is 'DiffuMamba.' This inconsistency is confusing and must be resolved in the revised version.
- [§4.3] The text says DiffuApriel-H+MLP gives 'roughly a 2% perplexity reduction' at 1.3B. From Table 2, 25.01→22.89 is a ~8.5% relative reduction and 22.72→20.17 is ~11.2% relative. The reported number is arithmetically wrong.
- [§5] The concluding limitation — 'performance advantages diminish on short contexts' — is not supported by any presented data. No short-context quality or throughput results are shown. Either add supporting analysis or remove the claim.
- [Tables 3–5] No error bars, seeds, or repeated-run statistics are reported for any perplexity or accuracy table. For a paper whose central claim is 'consistently outperform,' at least one repeated-seed evaluation or a statement about training variance is needed. This is particularly relevant in Table 4, where several differences are only 2–4 points.
- [Eq. (1)] The notation δ_{x_i^0} and δ_{MASK} is undefined; these are point masses on the token and [MASK], respectively. Please define them explicitly.
Circularity Check
No significant circularity: theoretical derivation is self-contained; empirical claims are measured against an external baseline, with minor self-citation burden and abstract overclaims.
full rationale
I walked the derivation chain: the masked diffusion objective (Eq. 2) is standard; the BiMamba denoiser (Eqs. 4-6) is a new architecture; the output distribution (Eq. 11) and loss are the same as MDM, so 'preserving the probabilistic structure' is a design property, not a circular derivation. The linear-time claim follows from standard O(B L d) vs O(B L^2 d) complexity and is supported by wall-clock measurements (Figure 2), not by fitted constants. The quality tables are empirical comparisons against an independently trained DiffuTran baseline; no 'prediction' is algebraically forced by a fitted parameter. Self-citations (Radhakrishna et al. 2025a,b; Ostapenko et al. 2025; Fathi et al. 2025) are motivational or taxonomic and are not load-bearing: no uniqueness theorem or ansatz is imported from them. I flag two non-circular problems: (1) the abstract's 'up to 8.2x and 4.3x' and 'cache-efficient block diffusion ... emerges as the only strategy that scales linearly' are unsupported by the body, which reports 4.4x/2.6x and says 'We leave this direction for future work'; (2) Table 1's parameter mismatch (1.6B/1.5B vs 1.3B) confounds the architecture-quality claim, since equalizing parameter counts is claimed but not achieved ('only difference lies in the internal mixer architecture'). These are correctness/validity concerns, not circularity. Score 2 reflects the minor self-citation burden only.
Axiom & Free-Parameter Ledger
free parameters (4)
- K (attention insertion period in hybrid) =
5
- Number of decoding steps S =
128
- Mamba state dimension d_state =
128
- KV-cache block size for DiffuTran+KV baseline =
32 (2 denoising steps per block)
axioms (5)
- standard math The masked-diffusion training objective (Eq. 2) is a valid reweighted variational bound for absorbing-state discrete diffusion.
- domain assumption The MCMC approximate reverse process (Eq. 3) yields valid samples with 128 decoding steps.
- ad hoc to paper Additive fusion of forward and backward scans (Eq. 6) provides a symmetric bidirectional context for denoising.
- domain assumption Wall-clock throughput at batch size 1, bf16, H100, with CUDA graphs is the right efficiency measure.
- domain assumption Chinchilla and Quokka compute budgets are adequately matched across the compared models.
read the original abstract
Diffusion language models (DLMs) have emerged as a promising alternative to autoregressive (AR) generation, yet their reliance on Transformer backbones limits inference efficiency due to quadratic attention or KV-cache overhead. We introduce DiffuMamba, a masked diffusion language model built on a bidirectional Mamba backbone that combines the diffusion objective with linear-time sequence modeling, and DiffuMamba-H, a hybrid variant with interleaved attention. Across scales up to 1.3B parameters, our models match Transformer-based diffusion in downstream performance while achieving up to 8.2x and 4.3x higher inference throughput, respectively, on long sequences. We further present a systematic analysis of inference efficiency across modern DLM variants combining asymptotic complexity with empirical measurements. Notably, cache-efficient block diffusion with Mamba mixers emerges as the only strategy that scales linearly with sequence length and achieves the strongest performance across all baselines, suggesting a promising direction for future diffusion-based generation systems.
Figures
Reference graph
Works this paper leans on
-
[1]
Josh Achiam, Steven Adler, Sandhini Agarwal, Lama Ahmad, Ilge Akkaya, Florencia Leoni Aleman, Diogo Almeida, Janko Altenschmidt, Sam Altman, Shyamal Anadkat, et al. Gpt-4 technical report. arXiv preprint arXiv:2303.08774,
-
[5]
As expected, the absolute scores remain modest at this scale, reflecting the difficulty of these reasoning and knowledge-intensive benchmarks at 1.3B scale. Nonetheless, a consistent trend emerges across all tasks:DiffuAprieland DiffuApriel-Hclearly outperformDiffuTranby 3.3 points on average, indicating that linear-time state space modeling provides a st...
2018
-
[7]
A cheaper and better diffusion language model with soft-masked noise
Jiaao Chen, Aston Zhang, Mu Li, Alex Smola, and Diyi Yang. A cheaper and better diffusion language model with soft-masked noise. InProceedings of the 2023 Conference on Empirical Methods in Natural Language Processing,
2023
-
[8]
Think you have solved question answering? try arc, the ai2 reasoning challenge
Peter Clark, Isaac Cowhey, Oren Etzioni, Tushar Khot, Ashish Sabharwal, Carissa Schoenick, and Oyvind Tafjord. Think you have solved question answering? try arc, the ai2 reasoning challenge. arXiv preprint arXiv:1803.05457,
-
[10]
10 Soham De, Samuel L Smith, Anushan Fernando, Aleksandar Botev, George Cristian-Muraru, Albert Gu, Ruba Haroun, Leonard Berrada, Yutian Chen, Srivatsan Srinivasan, et al. Griffin: Mix- ing gated linear recurrences with local attention for efficient language models.arXiv preprint arXiv:2402.19427,
-
[11]
Mamba: Linear-time sequence modeling with selective state spaces
Albert Gu and Tri Dao Goel. Mamba: Linear-time sequence modeling with selective state spaces. arXiv preprint arXiv:2312.00752,
-
[12]
Training compute-optimal large language models.arXiv preprint arXiv:2203.15556,
Jordan Hoffmann, Sebastian Borgeaud, Arthur Mensch, Elena Buchatskaya, Trevor Cai, Eliza Rutherford, Diego de Las Casas, Lisa Anne Hendricks, Johannes Welbl, Aidan Clark, et al. Training compute-optimal large language models.arXiv preprint arXiv:2203.15556,
-
[13]
Vm-ddpm: Vision mamba diffusion for medical image synthesis.arXiv preprint arXiv:2405.05667,
Zhihan Ju and Wanting Zhou. Vm-ddpm: Vision mamba diffusion for medical image synthesis.arXiv preprint arXiv:2405.05667,
-
[14]
Mercury: Ultra-fast language models based on diffusion.arXiv preprint arXiv:2506.17298,
Inception Labs, Samar Khanna, Siddhant Kharbanda, Shufan Li, Harshit Varma, Eric Wang, Sawyer Birnbaum, Ziyang Luo, Yanis Miraoui, Akash Palrecha, et al. Mercury: Ultra-fast language models based on diffusion.arXiv preprint arXiv:2506.17298,
-
[15]
Guanghao Li, Zhihui Fu, Min Fang, Qibin Zhao, Ming Tang, Chun Yuan, and Jun Wang. Diffuspec: Unlocking diffusion language models for speculative decoding.arXiv preprint arXiv:2510.02358, 2025a. Jeffrey Li, Alex Fang, Georgios Smyrnis, Maor Ivgi, Matt Jordan, Samir Yitzhak Gadre, Hritik Bansal, Etash Guha, Sedrick Scott Keh, Kushal Arora, et al. Datacomp-l...
-
[16]
A survey on diffusion language models
Tianyi Li, Mingda Chen, Bowei Guo, and Zhiqiang Shen. A survey on diffusion language models. arXiv preprint arXiv:2508.10875, 2025b. 11 Opher Lieber, Barak Lenz, Hofit Bata, Gal Cohen, Jhonathan Osin, Itay Dalmedigos, Erez Safahi, Shaked Meirom, Yonatan Belinkov, Shai Shalev-Shwartz, et al. Jamba: A hybrid transformer- mamba language model.arXiv preprint ...
-
[17]
Jian Liu, Leyang Cui, Hanmeng Liu, Dandan Huang, Yile Wang, and Yue Zhang. Logiqa: A challenge dataset for machine reading comprehension with logical reasoning.arXiv preprint arXiv:2007.08124,
Pith/arXiv arXiv 2007
-
[18]
Xiao Liu, Chenxu Zhang, Fuxiang Huang, Shuyin Xia, Guoyin Wang, and Lei Zhang. Vision mamba: A comprehensive survey and taxonomy.IEEE Transactions on Neural Networks and Learning Systems, 2025a. Yangzhou Liu, Yue Cao, Hao Li, Gen Luo, Zhe Chen, Weiyun Wang, Xiaobo Liang, Biqing Qi, Lijun Wu, Changyao Tian, et al. Sequential diffusion language models.arXiv...
-
[19]
dkv-cache: The cache for diffusion language models.arXiv preprint arXiv:2505.15781,
Xinyin Ma, Runpeng Yu, Gongfan Fang, and Xinchao Wang. dkv-cache: The cache for diffusion language models.arXiv preprint arXiv:2505.15781,
-
[21]
Training optimal large diffusion language models.arXiv preprint arXiv:2510.03280,
Jinjie Ni, Qian Liu, Chao Du, Longxu Dou, Hang Yan, Zili Wang, Tianyu Pang, and Michael Qizhe Shieh. Training optimal large diffusion language models.arXiv preprint arXiv:2510.03280,
-
[22]
Large language diffusion models.arXiv preprint arXiv:2502.09992,
Shen Nie, Fengqi Zhu, Zebin You, Xiaolu Zhang, Jingyang Ou, Jun Hu, Jun Zhou, Yankai Lin, Ji- Rong Wen, and Chongxuan Li. Large language diffusion models.arXiv preprint arXiv:2502.09992,
-
[23]
Apriel-h1: Towards efficient enterprise reasoning models.arXiv preprint arXiv:2511.02651,
Oleksiy Ostapenko, Luke Kumar, Raymond Li, Denis Kocetkov, Joel Lamy-Poirier, Shruthan Rad- hakrishna, Soham Parikh, Shambhavi Mishra, Sebastien Paquet, Srinivas Sunkara, et al. Apriel-h1: Towards efficient enterprise reasoning models.arXiv preprint arXiv:2511.02651,
-
[24]
Jingyang Ou, Shen Nie, Kaiwen Xue, Fengqi Zhu, Jiacheng Sun, Zhenguo Li, and Chongxuan Li. Your absorbing discrete diffusion secretly models the conditional distributions of clean data.arXiv preprint arXiv:2406.03736,
-
[25]
Han Peng, Peiyu Liu, Zican Dong, Daixuan Cheng, Junyi Li, Yiru Tang, Shuo Wang, and Wayne Xin Zhao. How efficient are diffusion language models? a critical examination of efficiency evaluation practices.arXiv preprint arXiv:2510.18480,
-
[27]
From s4 to mamba: A comprehensive survey on structured state space models
Shriyank Somvanshi, Md Monzurul Islam, Mahmuda Sultana Mimi, Sazzad Bin Bashar Polock, Gaurab Chhetri, and Subasish Das. From s4 to mamba: A comprehensive survey on structured state space models. InarXiv:2503.18970,
-
[28]
Dim: Diffusion mamba for efficient high-resolution image synthesis.arXiv preprint arXiv:2405.14224,
Yao Teng, Yue Wu, Han Shi, Xuefei Ning, Guohao Dai, Yu Wang, Zhenguo Li, and Xihui Liu. Dim: Diffusion mamba for efficient high-resolution image synthesis.arXiv preprint arXiv:2405.14224,
-
[29]
Llama: Open and efficient foundation language models.arXiv preprint arXiv:2302.13971, 2023a
Hugo Touvron, Thibaut Lavril, Gautier Izacard, Xavier Martinet, Marie-Anne Lachaux, Timothée Lacroix, Baptiste Rozière, Naman Goyal, Eric Hambro, Faisal Azhar, et al. Llama: Open and efficient foundation language models.arXiv preprint arXiv:2302.13971, 2023a. Hugo Touvron, Louis Martin, Kevin Stone, Peter Albert, Amjad Almahairi, Yasmine Babaei, Nikolay B...
-
[30]
A systematic analysis of hybrid linear attention.arXiv preprint arXiv:2507.06457, 2025a
Dustin Wang, Rui-Jie Zhu, Steven Abreu, Yong Shan, Taylor Kergan, Yuqi Pan, Yuhong Chou, Zheng Li, Ge Zhang, Wenhao Huang, et al. A systematic analysis of hybrid linear attention.arXiv preprint arXiv:2507.06457, 2025a. Jinhong Wang, Jintai Chen, Danny Chen, and Jian Wu. Lkm-unet: Large kernel vision mamba unet for medical image segmentation. In Marius Geo...
Pith/arXiv arXiv 2024
-
[31]
Xu Wang, Chenkai Xu, Yijie Jin, Jiachun Jin, Hao Zhang, and Zhijie Deng. Diffusion llms can do faster-than-ar inference via discrete diffusion forcing.arXiv preprint arXiv:2508.09192, 2025b. Chengyue Wu, Hao Zhang, Shuchen Xue, Zhijian Liu, Shizhe Diao, Ligeng Zhu, Ping Luo, Song Han, and Enze Xie. Fast-dllm: Training-free acceleration of diffusion llm by...
-
[32]
Visual mamba: A survey and new outlooks.arXiv preprint arXiv:2404.18861,
Rui Xu, Shu Yang, Yihui Wang, Yu Cai, Bo Du, and Hao Chen. Visual mamba: A survey and new outlooks.arXiv preprint arXiv:2404.18861,
-
[33]
Gated delta networks: Improving mamba2 with delta rule.arXiv preprint arXiv:2412.06464,
Songlin Yang, Jan Kautz, and Ali Hatamizadeh. Gated delta networks: Improving mamba2 with delta rule.arXiv preprint arXiv:2412.06464,
-
[34]
Dream 7b: Diffusion large language models.arXiv preprint arXiv:2508.15487,
Jiacheng Ye, Zhihui Xie, Lin Zheng, Jiahui Gao, Zirui Wu, Xin Jiang, Zhenguo Li, and Lingpeng Kong. Dream 7b: Diffusion large language models.arXiv preprint arXiv:2508.15487,
-
[36]
Fengqi Zhu, Rongzhen Wang, Shen Nie, Xiaolu Zhang, Chunwei Wu, Jun Hu, Jun Zhou, Jianfei Chen, Yankai Lin, Ji-Rong Wen, et al. Llada 1.5: Variance-reduced preference optimization for large language diffusion models.arXiv preprint arXiv:2505.19223,
-
[37]
Jingwei Zuo, Maksim Velikanov, Ilyas Chahed, Younes Belkada, Dhia Eddine Rhayem, Guillaume Kunsch, Hakim Hacid, Hamza Yous, Brahim Farhat, Ibrahim Khadraoui, et al. Falcon-h1: A family of hybrid-head language models redefining efficiency and performance.arXiv preprint arXiv:2507.22448,
-
[2013]
Aili Chen, Aonian Li, Bangwei Gong, Binyang Jiang, Bo Fei, Bo Yang, Boji Shan, Changqing Yu, Chao Wang, Cheng Zhu, et al. Minimax-m1: Scaling test-time compute efficiently with lightning attention.arXiv preprint arXiv:2506.13585,
-
[2015]
A survey on efficient inference for large language models.arXiv preprint arXiv:2404.14294,
Zixuan Zhou, Xuefei Ning, Ke Hong, Tianyu Fu, Jiaming Xu, Shiyao Li, Yuming Lou, Luning Wang, Zhihang Yuan, Xiuhong Li, et al. A survey on efficient inference for large language models.arXiv preprint arXiv:2404.14294,
-
[2017]
Can a suit of armor conduct electricity? a new dataset for open book question answering
Todor Mihaylov, Peter Clark, Tushar Khot, and Ashish Sabharwal. Can a suit of armor conduct electricity? a new dataset for open book question answering. InProceedings of the 2018 Conference on Empirical Methods in Natural Language Processing,
2018
-
[2019]
Apriel-nemotron- 15b-thinker.arXiv preprint arXiv:2508.10948, 2025a
Shruthan Radhakrishna, Soham Parikh, Gopal Sarda, Anil Turkkan, Quaizar V ohra, Raymond Li, Dhruv Jhamb, Kelechi Ogueji, Aanjaneya Shukla, Oluwanifemi Bamgbose, et al. Apriel-nemotron- 15b-thinker.arXiv preprint arXiv:2508.10948, 2025a. Shruthan Radhakrishna, Aman Tiwari, Aanjaneya Shukla, Masoud Hashemi, Rishabh Maheshwary, Shiva Krishna Reddy Malay, Jas...
-
[2020]
Language models are few-shot learners.Advances in neural information processing systems, 33:1877–1901,
Tom Brown, Benjamin Mann, Nick Ryder, Melanie Subbiah, Jared D Kaplan, Prafulla Dhariwal, Arvind Neelakantan, Pranav Shyam, Girish Sastry, Amanda Askell, et al. Language models are few-shot learners.Advances in neural information processing systems, 33:1877–1901,
1901
-
[2021]
Sangmin Bae, Bilge Acun, Haroun Habeeb, Seungyeon Kim, Chien-Yu Lin, Liang Luo, Junjie Wang, and Carole-Jean Wu. Hybrid architectures for language models: Systematic analysis and design insights.arXiv preprint arXiv:2510.04800,
-
[2023]
Marianne Arriola, Aaron Gokaslan, Justin T Chiu, Zhihan Yang, Zhixuan Qi, Jiaqi Han, Sub- ham Sekhar Sahoo, and V olodymyr Kuleshov. Block diffusion: Interpolating between autoregres- sive and diffusion language models.arXiv preprint arXiv:2503.09573,
-
[2024]
Tri Dao and Albert Gu. Transformers are ssms: Generalized models and efficient algorithms through structured state space duality.arXiv preprint arXiv:2405.21060,
-
[2025]
Trinayan Baruah, Kaustubh Shivdikar, Sara Prescott, and David Kaeli. Characterizing the behavior of training mamba-based state space models on gpus.arXiv preprint arXiv:2508.17679,
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.