igitman commited on
Commit
0aa1573
·
verified ·
1 Parent(s): fc61fb8

Add model card and Bias/Explainability/Privacy/Safety subcards (MR !102 f73d94f)

Browse files
Files changed (5) hide show
  1. README.md +191 -0
  2. bias.md +5 -0
  3. explainability.md +12 -0
  4. privacy.md +4 -0
  5. safety.md +6 -0
README.md ADDED
@@ -0,0 +1,191 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: other
4
+ license_name: openmdw-1.1
5
+ license_link: >-
6
+ https://openmdw.ai/license/1-1/
7
+ pipeline_tag: text-generation
8
+ language:
9
+ - en
10
+ base_model:
11
+ - nvidia/NVIDIA-Nemotron-3-Ultra-550B-A55B-BF16
12
+ base_model_relation: finetune
13
+ datasets:
14
+ - nvidia/Nemotron-Math-Proofs-v3-RL
15
+ tags:
16
+ - nvidia
17
+ - pytorch
18
+ - nemotron-3
19
+ - latent-moe
20
+ - mtp
21
+ track_downloads: true
22
+ ---
23
+ # Nemotron-3-Labs-Ultra-Math-RL
24
+
25
+ *Referred to as Nemotron-3-Ultra-RL in the technical report.*
26
+
27
+ ## Description
28
+
29
+ Nemotron-3-Labs-Ultra-Math-RL is a decoder-only transformer language model specialized for mathematical reasoning, trained to solve difficult mathematical problems and identify mistakes in proofs, and deployed as part of an ensemble system that achieved a gold-medal level score at the International Mathematical Olympiad 2026. Full details can be found at [Accompanying Tech Report](https://github.com/NVIDIA-NeMo/Skills/blob/main/recipes/nemotron-imo-tts/paper.pdf).
30
+ Nemotron-3-Labs-Ultra-Math-RL was developed by NVIDIA as a part of Nemotron.
31
+
32
+ This model is ready for commercial and non-commercial use.
33
+
34
+ ### License/Terms of Use
35
+
36
+ Governing Download Terms: Use of this model is governed by the [OpenMDW License Agreement, version 1.1](https://raw.githubusercontent.com/OpenMDW/OpenMDW/refs/heads/main/1.1/LICENSE.OpenMDW-1.1) (OpenMDW-1.1).
37
+
38
+ ### Deployment Geography
39
+
40
+ Global
41
+
42
+ ### Use Case
43
+
44
+ Researchers and developers focused on AI-driven mathematical reasoning and proof verification, aiming to advance open models for solving complex math problems and improving reasoning capabilities.
45
+
46
+ ### Release Date
47
+
48
+ **HuggingFace:** September 3, 2026 via https://huggingface.co/collections/nvidia/nemotron-labs-imo-2026
49
+
50
+ ## Reference(s)
51
+
52
+ * [Accompanying Tech Report](https://github.com/NVIDIA-NeMo/Skills/blob/main/recipes/nemotron-imo-tts/paper.pdf)
53
+ * [nvidia/NVIDIA-Nemotron-3-Ultra-550B-A55B-BF16](https://huggingface.co/nvidia/NVIDIA-Nemotron-3-Ultra-550B-A55B-BF16)
54
+ * [nvidia/Nemotron-Math-Proofs-v3-RL](https://huggingface.co/datasets/nvidia/Nemotron-Math-Proofs-v3-RL)
55
+ * [nvidia/Nemotron-IMO-Bench](https://huggingface.co/datasets/nvidia/Nemotron-IMO-Bench)
56
+ * Inference pipeline and submitted proofs: https://github.com/NVIDIA-NeMo/Skills/tree/main/recipes/nemotron-imo-tts
57
+ * RL training recipe: https://github.com/NVIDIA-NeMo/RL/blob/imo-26-ultra-v3/docs/guides/nemotron-3-ultra-imo.md
58
+
59
+ ## Model Architecture
60
+
61
+ **Architecture Type:** Mamba2-Transformer Hybrid Latent Mixture of Experts (LatentMoE) with Multi-Token Prediction (MTP)
62
+ **Network Architecture:** Nemotron Hybrid LatentMoE
63
+ **This model was developed based on [nvidia/NVIDIA-Nemotron-3-Ultra-550B-A55B-BF16](https://huggingface.co/nvidia/NVIDIA-Nemotron-3-Ultra-550B-A55B-BF16).**
64
+ **Number of model parameters:** 550B Total / 55B Active
65
+
66
+ ## Input
67
+
68
+ **Input Type(s):** Text
69
+ **Input Format(s):** String
70
+ **Input Parameters:** One-Dimensional (1D)
71
+ **Other Properties Related to Input:** Maximum context length up to 1M tokens
72
+
73
+ ## Output
74
+
75
+ **Output Type(s):** Text
76
+ **Output Format:** String
77
+ **Output Parameters:** One-Dimensional (1D)
78
+ **Other Properties Related to Output:** Maximum context length up to 1M tokens
79
+
80
+ Our AI models are designed and/or optimized to run on NVIDIA GPU-accelerated systems. By leveraging NVIDIA's hardware (e.g. GPU cores) and software frameworks (e.g., CUDA libraries), the model achieves faster training and inference times compared to CPU-only solutions.
81
+
82
+ ## Software Integration
83
+
84
+ **Runtime Engine(s):** vLLM
85
+ **Supported Hardware Microarchitecture Compatibility:**
86
+ * NVIDIA Blackwell
87
+ * NVIDIA Hopper
88
+
89
+ **Supported Operating System(s):** Linux
90
+
91
+ The integration of foundation and fine-tuned models into AI systems requires additional testing using use-case-specific data to ensure safe and effective deployment. Following the V-model methodology, iterative testing and validation at both unit and system levels are essential to mitigate risks, meet technical and functional requirements, and ensure compliance with safety and ethical standards before deployment.
92
+
93
+ ## Model Version(s)
94
+
95
+ Nemotron-3-Labs-Ultra-Math-RL v1
96
+
97
+ ## Deployment
98
+
99
+ The Ultra BF16 checkpoint is a frontier-scale model. The minimum recommended hardware is:
100
+
101
+ * **Single-node:** 8× B200 (≈1.5 TB aggregate HBM — fits BF16 weights plus KV cache with headroom)
102
+ * **Multi-node:** ≥8 GPUs across H100 / H200 / GB200 / GB300, orchestrated with Ray v2
103
+
104
+ For more detailed information, please see the [nvidia/NVIDIA-Nemotron-3-Ultra-550B-A55B-BF16](https://huggingface.co/nvidia/NVIDIA-Nemotron-3-Ultra-550B-A55B-BF16) model card and [this cookbook](https://github.com/NVIDIA-NeMo/Nemotron/blob/main/usage-cookbook/Nemotron-3-Ultra/vllm_cookbook.ipynb).
105
+
106
+ Recommended container: `vllm/vllm-openai:v0.22.0`.
107
+
108
+ ```shell
109
+ export MODEL_CKPT=PATH/TO/MODEL/CHECKPOINT
110
+ ```
111
+
112
+ **8× B200 single-node deployment:**
113
+
114
+ ```shell
115
+ docker run -d --name nemotron-ultra-vllm \
116
+ --gpus all \
117
+ --ipc=host \
118
+ --network=host \
119
+ --shm-size=16g \
120
+ --ulimit memlock=-1 \
121
+ --ulimit stack=67108864 \
122
+ -v $MODEL_CKPT:/model:ro \
123
+ -e VLLM_WORKER_MULTIPROC_METHOD=spawn \
124
+ -e SAFETENSORS_FAST_GPU=1 \
125
+ -e NVIDIA_TF32_OVERRIDE=1 \
126
+ -e VLLM_LOGGING_LEVEL=INFO \
127
+ vllm/vllm-openai:v0.22.0 \
128
+ /model \
129
+ --host 0.0.0.0 \
130
+ --port 8000 \
131
+ --served-model-name nvidia/Nemotron-3-Labs-Ultra-Math-RL \
132
+ --trust-remote-code \
133
+ --tensor-parallel-size 8 \
134
+ --enable-expert-parallel \
135
+ --dtype bfloat16 \
136
+ --max-model-len 262144 \
137
+ --gpu-memory-utilization 0.90 \
138
+ --max-num-seqs 16 \
139
+ --max-num-batched-tokens 32768 \
140
+ --enable-chunked-prefill \
141
+ --enable-prefix-caching \
142
+ --reasoning-parser nemotron_v3 \
143
+ --enable-auto-tool-choice \
144
+ --tool-call-parser qwen3_coder \
145
+ --mamba-ssm-cache-dtype float16 \
146
+ --mamba-backend flashinfer \
147
+ --enable-mamba-cache-stochastic-rounding \
148
+ --mamba-cache-philox-rounds 5 \
149
+ --speculative-config '{"method": "nemotron_h_mtp", "num_speculative_tokens": 5}' \
150
+ --model-loader-extra-config '{"enable_multithread_load": true, "num_threads": 96}'
151
+ ```
152
+
153
+ Context length defaults to 256k above. To use up to 1M, set `VLLM_ALLOW_LONG_MAX_MODEL_LEN=1` and `--max-model-len 1048576`.
154
+
155
+ ## Training, Testing, and Evaluation Datasets
156
+
157
+ ### Training Dataset
158
+
159
+ **Data Modality:** Text
160
+
161
+ **Text Training Data Size:** 9,597 proof-generation prompts (50,430,007 bytes)
162
+ **Data Collection Method by dataset:** Hybrid: Automated, manually-collected, Synthetic
163
+ **Labeling Method by dataset:** Hybrid: Automated, manually-labeled, Synthetic
164
+ **Properties (Quantity, Dataset Descriptions, Sensor(s)):** Nemotron-Math-Proofs-v3-RL is a long-form mathematical reasoning dataset for reinforcement learning. The release contains 9,597 proof-generation prompts. The dataset uses [NeMo Gym](https://github.com/NVIDIA-NeMo/gym)-compatible, single-turn user prompts derived from hard proof problems in the [AoPS](https://artofproblemsolving.com/) subset of [nvidia/Nemotron-Math-Proofs-v1](https://huggingface.co/datasets/nvidia/Nemotron-Math-Proofs-v1). The `train` split asks the policy to produce a rigorous solution and self-evaluation. Policy responses and realized rewards are generated during training and are not stored in the file. See [nvidia/Nemotron-Math-Proofs-v3-RL](https://huggingface.co/datasets/nvidia/Nemotron-Math-Proofs-v3-RL).
165
+
166
+ ### Testing Dataset
167
+
168
+ **Data Collection Method by dataset:** Hybrid: Automated, manually-collected, Synthetic<br>
169
+ **Labeling Method by dataset:** Hybrid: Automated, manually-labeled, Synthetic<br>
170
+ **Properties (Quantity, Dataset Descriptions, Sensor(s)):** This corpus comprises only benchmarks for assessing the quality of mathematical proof generation and verification.
171
+
172
+ ### Evaluation Dataset
173
+
174
+ **Benchmark Score:** As part of an ensemble system, this model achieved a gold-medal-level score at the International Mathematical Olympiad 2026. Additional evaluation results are available in the [Accompanying Tech Report](https://github.com/NVIDIA-NeMo/Skills/blob/main/recipes/nemotron-imo-tts/paper.pdf).
175
+
176
+ **Data Collection Method by dataset:** Hybrid: Automated, manually-collected, Synthetic<br>
177
+ **Labeling Method by dataset:** Hybrid: Automated, manually-labeled, Synthetic<br>
178
+ **Properties (Quantity, Dataset Descriptions, Sensor(s)):** This corpus comprises only benchmarks for assessing the quality of mathematical proof generation and verification.
179
+
180
+ ## Inference
181
+
182
+ **Acceleration Engine:** vLLM, PyTorch
183
+ **Hardware Requirements (GPU Architecture, Model):**
184
+ * NVIDIA Ampere - A100 80 GB
185
+ * NVIDIA Hopper - H100
186
+
187
+ ## Ethical Considerations
188
+
189
+ NVIDIA believes Trustworthy AI is a shared responsibility and we have established policies and practices to enable development for a wide array of AI applications. Developers should work with their internal model team to ensure this model meets requirements for the relevant industry and use case and addresses unforeseen product misuse.
190
+ For more detailed information on ethical considerations for this model, please see the Model Card++ [Explainability](./explainability.md), [Bias](./bias.md), [Safety & Security](./safety.md), and [Privacy](./privacy.md) Subcards.
191
+ Please report model quality, risk, security vulnerabilities or NVIDIA AI Concerns [here](https://www.nvidia.com/en-us/support/submit-security-vulnerability/).
bias.md ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ Field | Response
2
+ :---------------------------------------------------------------------------------------------------|:---------------
3
+ Participation considerations from adversely impacted groups [protected classes](https://calcivilrights.ca.gov/disputeresolution/protected-characteristics/) in model design and testing: | [None]
4
+ Measures taken to mitigate against unwanted bias: | Supervised Fine-Tuning (SFT) and Reinforcement Learning from Human Feedback (RLHF) employed to calibrate the model’s reasoning capabilities to maintain logical consistency and appropriate complexity when interacting with or interpreting data from diverse age demographics.
5
+ Bias Metric (If Measured): | BBQ Accuracy Scores in Ambiguous Contexts . https://github.com/nyu-mll/BBQ/
explainability.md ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Field | Response
2
+ :------------------------------------------------------------------------------------------------------|:---------------------------------------------------------------------------------
3
+ Intended Task/Domain: | Text generation, reasoning, and chat
4
+ Model Type: | Text-to-text Mamba2-Transformer Hybrid
5
+ Intended Users: | Generative AI creators working with conversational AI models and image content.
6
+ Output: | Text
7
+ List the groups (or group characteristics) for which this was tested to produce comparable outcomes: | Not Applicable
8
+ Technical Limitations & Mitigation: | This model performs particularly well in instruction following regimes, as such may be strongly influenced by untrusted inputs and should be paired with appropriate guardrails and data filtering to better align use-case behaviors when exposed to such data.
9
+ Verified to have met prescribed NVIDIA quality standards: | Yes
10
+ Performance Metrics: | Accuracy, Throughput, and User-side throughput
11
+ Potential Known Risks: | The model was optimized explicitly for instruction following and as such is more susceptible to prompt injection and jailbreaking in various forms as a result of its instruction tuning. This means that the model should be paired with additional rails or system filtering to limit exposure to instructions from malicious sources -- either directly or indirectly by retrieval (e.g. via visiting a website) -- as they may yield outputs that can lead to harmful, system-level outcomes up to and including remote code execution in agentic systems when effective security controls including guardrails are not in place. The model may generate answers that may be inaccurate, omit key information, include irrelevant or redundant text, or produce socially unacceptable or undesirable text, even if the prompt itself does not include anything explicitly offensive.
12
+ Licensing: | Use of this model is governed by the OpenMDW License Agreement, version 1.1 (OpenMDW-1.1).
privacy.md ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ Privacy Information
2
+ Nemotron-3-Labs-Ultra-Math-SFT/RL model was trained on large-scale publicly available data that may contain images, audio-video, and text relating to people. NVIDIA collected and used this data in compliance with applicable data protection and privacy laws. This model was not designed to derive insights or otherwise learn from any personal data contained in the datasets.
3
+ NVIDIA uses a combination of filters, data minimization techniques, and other guardrails to help prevent personal data from being recited by our models. We employ automated tools and data processing techniques during pre-training or training to identify and filter certain categories of personal data.
4
+ Please review NVIDIA's Privacy Policy for more information. https://www.nvidia.com/en-us/about-nvidia/privacy-policy/
safety.md ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ Field | Response
2
+ :---------------------------------------------------|:----------------------------------
3
+ Model Application Field(s): | Chat, Instruction Following, Chatbot Development, Code Generation, Reasoning, Customer Service
4
+ Describe the life critical impact (if present). | Not Applicable
5
+ Use Case Restrictions: | Use of this model is governed by the OpenMDW License Agreement, version 1.1 (OpenMDW-1.1).
6
+ Model and dataset restrictions: | The Principle of least privilege (PoLP) is applied limiting access for dataset generation and model development. Restrictions enforce dataset access during training, and dataset license constraints adhered to.