GO TO A SECTION

↑ ↓ to navigate · Enter to openExplore all sections →
FDEInterviews logo
arXiv preprint

Nemotron 3 Nano: Open, Efficient Mixture-of-Experts Hybrid Mamba-Transformer Model for Agentic Reasoning

NVIDIA's 31.6B model activates 3.2B per token and has only 6 attention layers among 52, so its KV cache is tiny. Fast long generation and 1M context; weaker on knowledge and multilingual tests.

Breakdown by FDE Interviews · 13 min read · Updated

About the original · 2025
AUTHORS
NVIDIA
FROM
NVIDIA
PUBLISHED
Dec 2025 · arXiv preprint
CITED BY
99Semantic Scholar · Oct 2026
IN THIS BREAKDOWN · 10 SECTIONS

You can read 17% of this breakdown.

01

The paper in one minute

Nemotron 3 Nano 30B-A3B is a 52-layer model that mixes three layer types. 23 are Mamba-2 layers, a state-space design that carries a fixed-size state instead of a growing cache. 23 are mixture-of-experts layers with 128 routed experts (6 active) and 2 shared ones. Only 6 are attention layers, each with 32 query heads sharing 2 key-value heads. The model has 31.6B parameters, 3.2B active per token (3.6B with embeddings), and no positional embeddings.

It was pre-trained on 25 trillion tokens in two phases (23.5T diverse, then 1.5T high quality), with a 121B-token long-context phase mixing 512K and 4K sequences. Post-training has three parts: SFT on 18M samples, with reasoning removed from 10% to teach an on/off switch and cut short in 3% to teach budget control; reinforcement learning on all environments at once (math, code, question answering, JSON schemas, instruction following, long context, tool use); and RLHF with a 235B generative reward model. A final FP8 version keeps the attention layers and the Mamba layers feeding them in bf16.

On one H200, with 8K tokens in and 16K out, it produces 3.3 times the output tokens per second of Qwen3-30B-A3B-Thinking-2507 and 2.2 times GPT-OSS-20B. It leads both on chat (Arena-Hard-V2), instruction following (IFBench), coding (LiveCodeBench v6, SWE-Bench) and RULER up to 1M tokens. It trails Qwen3 on MMLU-Pro and on AA-LCR, and both rivals on multilingual MMLU-ProX. Weights, most of the data and the training recipe are released.

THE IDEA, VISUALLY
52 LAYERS, ONLY 6 OF THEM ATTENTION short = Mamba-2 (23) tall = MoE (23) accent = attention (6) MoE: 128 routed experts, 6 active + 2 shared; 31.6B total, 3.2B active attention: 32 query heads sharing 2 KV heads of dimension 128 no positional embeddings, no biases, untied embeddings KV CACHE PER TOKEN (FROM TABLE 1) Nemotron 3 Nano, bf16 6 KiB Nemotron 3 Nano, fp8 3 KiB same, if all 52 were attention 52 KiB 1M-token session: about 6 GiB of KV in bf16, plus a fixed Mamba state of about 23 MiB per sequence POST-TRAINING SFT on 18M samples; reasoning stripped from 10% (on/off switch) and cut short in 3% (budget control) RL on all environments at once: math, code, tools, JSON, long context RLHF with a 235B generative reward model; FP8 PTQ keeps attention and the 6 Mamba layers before it in bf16 VS QWEN3-30B-A3B-THINKING AND GPT-OSS-20B Nano Qwen3 GPT-OSS Arena-Hard-V2 avg 67.65 57.80 48.55 IFBench 71.51 51.00 65.00 LiveCodeBench v6 68.25 66.00 61.00 SWE-Bench (OpenHands) 38.76 22.00 34.00 RULER @ 1M 86.34 77.50 n/a TauBench V2 avg 49.04 47.70 47.50 MMLU-Pro 78.30 80.90 75.00 AA-LCR (long, reasoning on) 35.85 59.00 34.00 MMLU-ProX (multilingual) 59.50 77.60 69.10 some rival scores come from ArtificialAnalysis or NVIDIA's runs THROUGHPUT, 8K IN / 16K OUT, ONE H200 Nemotron 3 Nano (FP8) 3.3x GPT-OSS-20B (mxfp4) 1.5x Qwen3-30B-A3B (FP8) 1.0x relative output tokens/s/GPU, best of vLLM and TRT-LLM FP8 VS BF16 (TABLE 4) median accuracy kept ~99%; AIME25 89.06 → 87.71, Tau airline 48.00 → 44.79, HLE with tools 15.48 → 14.27
Mamba-2 layers replace most attention, so only 6 layers keep a KV cache. That is why long outputs and million-token contexts are cheap here, and why the model leads on generation-heavy throughput.
Research notes: the problem, approach and results
3 KiB

KV cache per token in FP8, computed from Table 1: 6 attention layers with 2 KV heads of dimension 128

3.3x

output throughput versus Qwen3-30B-A3B-Thinking-2507 at 8K input and 16K output on one H200

8.33% → 0.7%

rate of calling tools that were never declared, on GPQA, after a 50-step DPO run in the appendix

THE PROBLEM

Agentic and reasoning workloads generate long outputs and carry long contexts, and in a standard Transformer the KV cache grows with every token in every layer. That cache, not the weights, limits how many requests a GPU can serve.

THE IDEA

Combine Mamba-2 layers, which keep a fixed-size state, with just 6 grouped-query attention layers and 23 mixture-of-experts layers (6 of 128 experts active). Pre-train on 25T tokens, then train reasoning on/off and budget control in SFT and run RL across many environments at once.

THE RESULT

At 8K input and 16K output on one H200, throughput is 3.3 times Qwen3-30B-A3B-Thinking and 2.2 times GPT-OSS-20B. It leads them on Arena-Hard-V2, IFBench, LiveCodeBench and SWE-Bench, scores 86.34 on RULER at 1M tokens, and trails on MMLU-Pro, AA-LCR and multilingual MMLU-ProX.

02

Why it matters to an FDE

For agent and reasoning workloads, serving cost is set by how many concurrent requests fit on a GPU, and that is set by the KV cache. A reasoning model that writes 16K tokens per answer, or an agent that carries 100K tokens of history, fills memory with cache long before it runs out of compute.

Nemotron 3 Nano attacks that directly. With 6 attention layers and 2 KV heads each, the cache is about 3 KiB per token in FP8, computed from the report's Table 1; the same shape in all 52 layers would be 8.7 times larger. A million-token session needs about 3 GiB of cache. This is the mechanism behind its throughput lead on generation-heavy work, and why 1M-token context is practical on one GPU.

It is also a reference point for choosing a mid-size open model. It wins on tool use, coding and chat, and its reasoning switch and budget control were trained in rather than bolted on. It is weaker where knowledge breadth matters: MMLU-Pro, and especially multilingual MMLU-ProX (59.5 against 77.6 for Qwen3). A customer serving English agentic workloads is its sweet spot; a multilingual knowledge assistant is not.

THE FULL BREAKDOWN

Understand the mechanism. Try it on a real problem.

Unlock this breakdown and 149 more papers.

Premium opens every paper in full, plus every answer, concept, course and guide on the site. One payment for six months of access.

6 months · One payment · No auto-renewal

Get Premium · ₹2,000 / $25
  • ◇1 architecture diagram
  • </>1 runnable code experiment
  • ▦2 results tables
  • ?5 interview questions with answers
  1. 03Key takeaways

    Memory arithmetic, what the post-training adds, FP8, tool hallucination and three misreadings

    Free account
  2. 04How it works

    The layer pattern, pre-training and long context, SFT controls, multi-environment RL and FP8

    Premium
  3. 05The numbers that matter

    Base model, post-trained comparison, throughput setup, FP8 recovery, prompt sensitivity and DPO

    Premium
  4. 06Try it yourself

    Count the layers, size the KV cache and GPU capacity, and test the table claims against prompt noise

    Premium
  5. 07Where it breaks

    Knowledge and multilingual gaps, reasoning-on long context, mixed comparison sources and slips

    Premium
  6. 08Using it on a deployment

    Matching workloads to the hybrid, sizing GPUs, FP8 checks and controlling reasoning and tools

    Premium
  7. 09Interview angles

    Five questions this report prepares you for, with answer hints

    Premium
Back to the paper library →