Open, Efficient Mixture-of-Experts Hybrid Mamba-Transformer for Agentic Reasoning
NVIDIA
Read the Paper on arXivNemotron 3 Super is NVIDIA's open-weight large language model that combines three architectural innovations — Hybrid Mamba-Attention, LatentMoE, and Multi-Token Prediction — into a single 120B total parameter model with only 12B active parameters per forward pass (12.7B including embeddings). The model uses a vocabulary of 131,072 tokens, a hidden dimension of , and spans 88 layers with a mix of Mamba-2 SSM blocks and grouped-query attention layers.
The model achieves comparable or better accuracy than GPT-OSS-120B (DeepSeek-V3-0324) and Qwen3.5-122B across reasoning, coding, agentic, and long-context benchmarks, while delivering dramatically better serving efficiency: 2.2x higher throughput than GPT-OSS-120B and 7.5x higher throughput than Qwen3.5-122B on realistic workloads (8K input / 64K output tokens) on NVIDIA B300 GPUs. This efficiency advantage comes from three sources: (1) Mamba-2 blocks eliminate KV cache growth for the majority of layers, (2) LatentMoE reduces memory bandwidth per expert by 4x through latent projection, and (3) Multi-Token Prediction enables native speculative decoding with an average acceptance length of 3.45 tokens — no external draft model required.
Key architectural and training advances:
Nemotron 3 Super is fully open-sourced on HuggingFace — base, post-trained, FP8, and NVFP4 checkpoints — along with all training datasets (Nemotron-Pretraining-Specialized-v1.1 and Nemotron-Super-Post-Training-Data).
Click any topic to jump in
LatentMoE → hybrid SSM → NVFP4 pretrain → multi-env RL → deployment quantization compound into 3× efficiency.
Latent projection before expert compute makes wall-clock cost invariant to expert count, enabling 256 experts.
Interleaved SSM and attention layers give linear-context reasoning with 5× smaller KV cache at 128K.
Auxiliary heads predicting k future tokens densify supervision and enable free speculative decoding.
4-bit microscaled format halves memory and doubles tensor-core throughput with <0.3% loss gap.
Simultaneous RL across math/code/tool/chat with per-env rewards prevents single-domain reward hacking.
Per-layer FP4/FP8/INT8 search yields 4× throughput with <1% quality loss via error-budget optimization.
The complete Nemotron 3 Super architecture — from hybrid Mamba-Attention backbone through LatentMoE sparse computation and multi-token prediction to efficient quantized deployment. Each component addresses a specific bottleneck in serving efficiency while maintaining or improving model quality.
120B total parameters, 12B active per forward pass (12.7B including embeddings) — 10:1 sparsity ratio via LatentMoE expert selection
88-layer hybrid stack: periodic pattern of Mamba-2 SSM blocks (state dim 128, 128 heads, 8 groups, head dim 64) + GQA attention anchors (32 Q / 2 KV heads, head dim 128) + LatentMoE FFN layers (512 experts, top-22 in 1024-dim latent space)
LatentMoE: token projected d=4096 → l=1024, then routed to 22 of 512 experts operating in latent space — 4x less memory bandwidth per expert, exponentially larger combinatorial routing space
Multi-Token Prediction: 2 shared-weight auxiliary heads predicting future offsets — average acceptance length 3.45 tokens on SPEED-Bench (vs DeepSeek-R1's 2.70), shifting throughput-latency Pareto frontier to ~1000 TPS at D=3
Pre-trained on 25 trillion tokens in NVFP4 (E2M1 with 16-element micro-blocks): Phase 1 = 20T tokens diverse web/code/academic/multilingual, Phase 2 = 5T tokens high-quality PDFs/Wikipedia/curated
Post-trained via 5 stages: SFT (7M samples, 2-stage loss) → RLVR (21 environments, 37 datasets, async GRPO on up to 1K GPUs) → SWE-RL (end-to-end GitHub issues in Apptainer containers) → RLHF (GenRM on Qwen3-235B-Thinking) → MTP healing (frozen backbone, retrain speculative heads)
Quantized to FP8 (99.5% accuracy) and NVFP4 (99.8% accuracy) via AutoQuantize — NAS-inspired knapsack optimization for per-operator precision under 4.75-bit budget, with stochastic rounding for Mamba SSM cache
2.2x throughput vs GPT-OSS-120B and 7.5x vs Qwen3.5-122B on B300 GPUs (8K in / 64K out) — fully open-sourced: BF16, FP8, NVFP4 checkpoints + all training data on HuggingFace
Nemotron 3 Super composes four ideas in sequence: LatentMoE (hardware-aware sparsity) Mamba-2 + attention hybrid (linear context) NVFP4 pretraining (4-bit compute) multi-environment RL post-training. Each stage attacks a different bottleneck — compute, memory, precision, and alignment — giving a compounding efficiency gain over dense baselines.