nvidia/AceMath-RL-Nemotron-7B

Model

Introduction

27

4 commits

1 linked in READMEs

updated Apr 23, 2025

See the code

README

Introduction

aime24_accuracy

We’re thrilled to introduce AceMath-RL-Nemotron-7B, a math reasoning model trained entirely through reinforcement learning (RL), starting from the Deepseek-R1-Distilled-Qwen-7B. It delivers impressive results, achieving 69.0% Pass@1 accuracy on AIME 2024 (+13.5% gain) and 53.6% Pass@1 accuracy on AIME 2025 (+14.4% gain). Interestingly, this math-focused RL training also improves the model’s coding accuracy on LiveCodeBench, reaching 44.4% Pass@1 (+6.8% gain), demonstrating the generalization capabilities of scaled RL training.

We share our training recipe, training logs, and data curation details in our BLOG.

Results

We evaluate our model against competitive reasoning models of comparable size on AIME 2024, AIME 2025, and GPQA.

ModelAIME 2024
(AVG@64)
AIME 2025
(AVG@64)
GPQA-Diamond
(AVG@8)
DeepSeek-R1-Distill-Qwen-7B55.539.249.1
Light-R1-7B-DS59.144.349.4
AReaL-boba-RL-7B61.948.347.6
Llama-Nemotron-Nano-v1 (8B)63.847.154.1
Skywork-OR1-Math-7B-Preview69.852.3-
AceMath-RL-Nemotron-7B 🤗69.053.652.1

Additionally, we evaluate our models on additional math benchmarks and LiveCodeBench for a more comprehensive evaluation.

ModelGSM8K
(AVG@1)
MATH500
(AVG@4)
Minerva Math
(AVG@1)
GaoKao2023En
(AVG@1)
Olympiad Bench
(AVG@1)
College Math
(AVG@1)
ACM23
(AVG@5)
LiveCodeBench
(AVG@8)
DeepSeek-R1-Distill-Qwen-7B92.792.857.482.358.256.789.037.6
AceMath-RL-Nemotron-7B 🤗93.394.156.685.566.759.894.044.4

How to use

import torch
from transformers import AutoModelForCausalLM, AutoTokenizer

model_name = 'nvidia/AceMath-RL-Nemotron-7B'
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto", device_map="auto")

prompt = "Jen enters a lottery by picking $4$ distinct numbers from $S=\\{1,2,3,\\cdots,9,10\\}.$ $4$ numbers are randomly chosen from $S.$ She wins a prize if at least two of her numbers were $2$ of the randomly chosen numbers, and wins the grand prize if all four of her numbers were the randomly chosen numbers. The probability of her winning the grand prize given that she won a prize is $\\tfrac{m}{n}$ where $m$ and $n$ are relatively prime positive integers. Find $m+n$."
messages = [{"role": "user", "content": prompt}]

text = tokenizer.apply_chat_template(
    messages,
    tokenize=False,
    add_generation_prompt=True
)
model_inputs = tokenizer([text], return_tensors="pt").to("cuda")

generated_ids = model.generate(
    **model_inputs,
    max_new_tokens=32768,
    temperature=0.6,
    top_p=0.95
)
generated_ids = [
    output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
]

response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]

Usage Recommendations

  1. Don't include a system prompt; instead, place all instructions directly in the user prompt.
  2. We recommend using the following prompt format for math questions:
    <|begin▁of▁sentence|><|User|>{math_question}\nPlease reason step by step, and put your final answer within \boxed{}.<|Assistant|><think>\n

Correspondence to

Yang Chen (yachen@nvidia.com),
Zihan Liu (zihanl@nvidia.com),
Chankyu Lee (chankyul@nvidia.com),
Wei Ping (wping@nvidia.com)

License

Your use of this model is governed by the NVIDIA Open Model License.

Citation

@article{acemath2024,
  title={AceMath: Advancing Frontier Math Reasoning with Post-Training and Reward Modeling},
  author={Liu, Zihan and Chen, Yang and Shoeybi, Mohammad and Catanzaro, Bryan and Ping, Wei},
  journal={arXiv preprint},
  year={2024}
}
conversational
endpoints_compatible
math
nvidia
pytorch
qwen2
reasoning
reinforcement learning
safetensors
text-generation
text-generation-inference
transformers

Contributors

zihanliu

4 commits

nvidia/AceMath-RL-Nemotron-7B

Model

Introduction

27

4 commits

1 linked in READMEs

updated Apr 23, 2025

See the code

README

Introduction

aime24_accuracy

We’re thrilled to introduce AceMath-RL-Nemotron-7B, a math reasoning model trained entirely through reinforcement learning (RL), starting from the Deepseek-R1-Distilled-Qwen-7B. It delivers impressive results, achieving 69.0% Pass@1 accuracy on AIME 2024 (+13.5% gain) and 53.6% Pass@1 accuracy on AIME 2025 (+14.4% gain). Interestingly, this math-focused RL training also improves the model’s coding accuracy on LiveCodeBench, reaching 44.4% Pass@1 (+6.8% gain), demonstrating the generalization capabilities of scaled RL training.

We share our training recipe, training logs, and data curation details in our BLOG.

Results

We evaluate our model against competitive reasoning models of comparable size on AIME 2024, AIME 2025, and GPQA.

ModelAIME 2024
(AVG@64)
AIME 2025
(AVG@64)
GPQA-Diamond
(AVG@8)
DeepSeek-R1-Distill-Qwen-7B55.539.249.1
Light-R1-7B-DS59.144.349.4
AReaL-boba-RL-7B61.948.347.6
Llama-Nemotron-Nano-v1 (8B)63.847.154.1
Skywork-OR1-Math-7B-Preview69.852.3-
AceMath-RL-Nemotron-7B 🤗69.053.652.1

Additionally, we evaluate our models on additional math benchmarks and LiveCodeBench for a more comprehensive evaluation.

ModelGSM8K
(AVG@1)
MATH500
(AVG@4)
Minerva Math
(AVG@1)
GaoKao2023En
(AVG@1)
Olympiad Bench
(AVG@1)
College Math
(AVG@1)
ACM23
(AVG@5)
LiveCodeBench
(AVG@8)
DeepSeek-R1-Distill-Qwen-7B92.792.857.482.358.256.789.037.6
AceMath-RL-Nemotron-7B 🤗93.394.156.685.566.759.894.044.4

How to use

import torch
from transformers import AutoModelForCausalLM, AutoTokenizer

model_name = 'nvidia/AceMath-RL-Nemotron-7B'
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto", device_map="auto")

prompt = "Jen enters a lottery by picking $4$ distinct numbers from $S=\\{1,2,3,\\cdots,9,10\\}.$ $4$ numbers are randomly chosen from $S.$ She wins a prize if at least two of her numbers were $2$ of the randomly chosen numbers, and wins the grand prize if all four of her numbers were the randomly chosen numbers. The probability of her winning the grand prize given that she won a prize is $\\tfrac{m}{n}$ where $m$ and $n$ are relatively prime positive integers. Find $m+n$."
messages = [{"role": "user", "content": prompt}]

text = tokenizer.apply_chat_template(
    messages,
    tokenize=False,
    add_generation_prompt=True
)
model_inputs = tokenizer([text], return_tensors="pt").to("cuda")

generated_ids = model.generate(
    **model_inputs,
    max_new_tokens=32768,
    temperature=0.6,
    top_p=0.95
)
generated_ids = [
    output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
]

response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]

Usage Recommendations

  1. Don't include a system prompt; instead, place all instructions directly in the user prompt.
  2. We recommend using the following prompt format for math questions:
    <|begin▁of▁sentence|><|User|>{math_question}\nPlease reason step by step, and put your final answer within \boxed{}.<|Assistant|><think>\n

Correspondence to

Yang Chen (yachen@nvidia.com),
Zihan Liu (zihanl@nvidia.com),
Chankyu Lee (chankyul@nvidia.com),
Wei Ping (wping@nvidia.com)

License

Your use of this model is governed by the NVIDIA Open Model License.

Citation

@article{acemath2024,
  title={AceMath: Advancing Frontier Math Reasoning with Post-Training and Reward Modeling},
  author={Liu, Zihan and Chen, Yang and Shoeybi, Mohammad and Catanzaro, Bryan and Ping, Wei},
  journal={arXiv preprint},
  year={2024}
}
conversational
endpoints_compatible
math
nvidia
pytorch
qwen2
reasoning
reinforcement learning
safetensors
text-generation
text-generation-inference
transformers

Contributors

zihanliu

4 commits