In-Place Indic Tokenizer Expansion and Continual Pretraining of Nemotron-3-Nano-30B-A3B
Abstract
We adapted Nemotron-3-Nano-30B-A3B for eight Indian languages without changing its backbone architecture. We expanded the vocabulary in place from 131,072 to 163,072 entries using continued byte-pair encoding and AdaptMix rebalancing, reducing tokens per word by 38.6%. We initialized new input and output rows by averaging their original constituent rows, then trained them on 90% Indic and 10% English text before updating the backbone. Continual pretraining then progressed through broad training, high-quality annealing, and long-context training over approximately 1.27 trillion tokens. The annealed checkpoint gained 12.9 points on the Indic MILU benchmark but lost 3.4 on English MMLU. We also report practical negative results. Direct token addition underperformed continued byte-pair encoding. AdamW weight decay changed rows intended to stay fixed. Perplexity on an Indic-heavy holdout did not select the learning rate that best preserved English performance.
Figures & tables
| Benchmark | Base | Broad | Anneal |
|---|---|---|---|
| MILU [ 12 ] | 40.63 | 53.54 | 53.52 |
| ARC-Challenge-Indic [ 30 ] | 35.23 | 48.20 | 48.20 |
| IndicGLUE-selected [ 14 ] | 52.12 | 57.86 | 57.22 |
| IndicMMLU-Pro [ 15 ] | 14.44 | 20.03 | 20.02 |
| BharatMath | 29.99 | 41.06 | 39.70 |
| Indic BPB overall ( ) [ 13 ] | 0.8453 | 0.6189 | 0.6132 |
Appendix figures & tables19 assets
Supplementary material from the paper’s appendix.
Appendix
| Run | Extension | Mean fertility | Notes |
|---|---|---|---|
| Base Nemotron | None | 3.5472 | Original tokenizer |
| Direct add_tokens | 32K | 2.4815 | 14.3% above continued BPE |
| Continued BPE | 32K | 2.1715 | Used in subsequent experiments |
| Raw / NFKC / NFKC+regex | 32K | 2.1716–2.1823 | Similar fertility across variants |
| TOK-2 | 32K | 2.1773 | Selected for CPT |
| TOK-3 | 64K | 2.0996 | Lowest mean, not used for CPT |
| Field | Value |
|---|---|
| Initialization | Checkpoint after warmup, vocabulary 163,072 |
| Precision | BF16 |
| Sequence length | 8,192 (broad and anneal), 131,072 (long context) |
| Global / micro batch | 2,048 / 1 |
| Parallelism | TP=1, EP=8, PP=1, remaining ranks use DP |
| Optimizer | Distributed AdamW |
| Language | Base | 120B CPT | |
|---|---|---|---|
| Kannada | 32.8 | 47.2 | 14.4 |
| Tamil | 34.2 | 46.6 | 12.4 |
| Gujarati | 34.4 | 46.5 | 12.2 |
| Telugu | 35.8 | 47.9 | 12.1 |
| Malayalam | 35.5 | 45.7 | 10.1 |
| Bengali | 39.4 | 47.7 | 8.4 |
| Bucket | Weight | Shards | Pool tokens |
|---|---|---|---|
| Indic long documents | 0.22 | 9 | 22.0B |
| Indic high-quality | 0.15 | 132 | 15.0B |
| English long documents | 0.20 | 7 | 20.0B |
| Mathematics and STEM | 0.14 | 1 | 14.0B |
| Long-context supervised data | 0.10 | 3 | 10.0B |
| Code | 0.08 | 1 | 8.0B |
| Benchmark | Base | Broad | Anneal |
|---|---|---|---|
| MILU [ 12 ] | 40.63 | 53.54 | 53.52 |
| ARC-Challenge-Indic [ 30 ] | 35.23 | 48.20 | 48.20 |
| Belebele [ 20 ] | 37.49 | 42.17 | 42.27 |
| BharatMath | 29.99 | 41.06 | 39.70 |
| Global-MMLU-Lite [ 21 ] | 32.50 | 36.17 | 36.00 |
| IndicGLUE-selected [ 14 ] | 52.12 | 57.86 | 57.22 |
| Metric or context length | Score (%) |
|---|---|
| Mean RULER recall (55 slices) | 76.7 |
| Mean QA exact match (10 slices) | 45.7 |
| Recall at 8,192 tokens | 96.6 |
| Recall at 16,384 tokens | 90.6 |
| Recall at 32,768 tokens | 82.4 |
| Recall at 65,536 tokens | 67.3 |
| Stage | GPU-hours |
|---|---|
| CPT broad ( 974B) | 25,147 |
| CPT annealing (195B) | 4,988 |
| CPT long context ( 100B) | 4,977 |
| CPT total | 35,112 |
| Field | Value |
|---|---|
| Total / active parameters | 31.75B / 3.31B |
| Layers | 52 (23 Mamba-2, 23 MoE, 6 GQA) |
| Model dim / heads / KV heads | 2,688 / 32 / 2 |
| MoE experts (routed / active / shared) | 128 / 6 / 1 |
| Vocabulary / max sequence | 163,072 / 131,072 |
| Position encoding | none (NoPE) |
| Language | Upstream | Expanded | Change |
|---|---|---|---|
| Hindi | 2.05 | 1.66 | 19.2% |
| Bengali | 3.06 | 2.14 | 30.0% |
| Gujarati | 3.61 | 1.91 | 47.1% |
| Kannada | 3.86 | 2.34 | 39.5% |
| Malayalam | 4.97 | 2.77 | 44.3% |
| Marathi | 3.10 | 1.94 | 37.4% |
| Bucket | Weight | Target tokens |
|---|---|---|
| Reasoning | 22% | 62.7B |
| Indic long documents | 15% | 42.8B |
| Math & STEM | 14% | 39.9B |
| Knowledge retention | 12% | 34.2B |
| English long documents | 11% | 31.4B |
| Code | 9% | 25.7B |
Language Model Pretraining
Language model pretraining teaches models language patterns and knowledge from large text corpora before they are adapted to specific tasks.
See all 441 papers