
| Model | AIME 2024 (avg@64) | AIME 2025 (avg@64) | LCB v5 (avg@8) | LCB v6 (avg@8) |
|---|---|---|---|---|
| QwQ-32B | 79.5 | 65.8 | 63.4 | - |
| DeepSeek-R1-671B | 79.8 | 70.0 | 65.9 | - |
| Llama-Nemotron-Ultra-253B | 80.8 | 72.5 | 66.3 | - |
| o3-mini (medium) | 79.6 | 76.7 | 67.4 | - |
| Light-R1-7B | 59.1 | 44.3 | 40.6 | 36.4 |
| Light-R1-14B | 74 | 60.2 | 57.9 | 51.5 |
| DeepCoder-14B (32K Inference) | 71 | 56.1 | 57.9 | 50.4 |
| OpenMath-Nemotron-7B | 74.8 | 61.2 | - | - |
| OpenCodeReasoning-Nemotron-7B | - | - | 51.3 | 46.1 |
| Llama-Nemotron-Nano-8B-v1 | 61.3 | 47.1 | 46.6 | 46.2 |
| DeepSeek-R1-Distilled-Qwen-7B | 55.5 | 39.0 | 37.6 | 34.1 |
| DeepSeek-R1-Distilled-Qwen-14B | 69.7 | 50.2 | 53.1 | 47.9 |
| DeepSeek-R1-Distilled-Qwen-32B | 72.6 | 54.9 | 57.2 | - |
| AceReason-Nemotron-7B 🤗 | 69.0 | 53.6 | 51.8 | 44.1 |
| AceReason-Nemotron-14B 🤗 | 78.6 | 67.4 | 61.1 | 54.9 |
1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4model_name = 'nvidia/AceReason-Nemotron-7B'
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto", device_map="auto")
7
8prompt = "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$."
9messages = [{"role": "user", "content": prompt}]
10
11text = tokenizer.apply_chat_template(
12 messages,
13 tokenize=False,
14 add_generation_prompt=True
15)
16model_inputs = tokenizer([text], return_tensors="pt").to("cuda")
17
18generated_ids = model.generate(
19 **model_inputs,
20 max_new_tokens=32768,
21 temperature=0.6,
22 top_p=0.95
23)
24generated_ids = [
25 output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
26]
27
28response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]1question = "" # code question
2starter_code = "" # starter code function header
3
4code_instruction_nostartercode = """Write Python code to solve the problem. Please place the solution code in the following format:\n```python\n# Your solution code here\n```"""
5code_instruction_hasstartercode = """Please place the solution code in the following format:\n```python\n# Your solution code here\n```"""
6if starter_code != "":
7 question += "\n\n" + "Solve the problem starting with the provided function header.\n\nFunction header:\n" + "```\n" + starter_code + "\n```"
8 question += "\n\n" + code_instruction_hasstartercode
9else:
10 question += "\n\n" + code_instruction_nostartercode
11
12final_prompt = "<|User|>" + question + "<|Assistant|><think>\n"@article{chen2025acereason,
title={AceReason-Nemotron: Advancing Math and Code Reasoning through Reinforcement Learning},
author={Chen, Yang and Yang, Zhuolin and Liu, Zihan and Lee, Chankyu and Xu, Peng and Shoeybi, Mohammad and Catanzaro, Bryan and Ping, Wei},
journal={arXiv preprint arXiv:2505.16400},
year={2025}
}