reasoning-gym

给AI推理训练提供可验证奖励的现成环境,直接上手RL

reasoning-gym 是一个为强化学习(RL)提供可验证奖励(verifiable rewards)的推理环境集合,旨在解决大语言模型(LLM)在推理任务中缺乏高质量、可自动评估的训练环境的问题。它提供了超过40种算法/数学/逻辑推理任务,如数独、国际象棋谜题、逻辑推理、代码执行等,每个任务都内置了可验证的奖励函数,使得RL训练能够基于客观结果而非人工反馈。项目采用Python编写,接口简洁,易于集成到现有的RL训练流程中,并支持与Hugging Face等生态协同。核心能力包括:任务生成、答案验证、环境标准化,以及可扩展的任务注册机制。该工作已被NeurIPS 2025接收为Spotlight,表明其学术价值。

开源 free 开源模型
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 1512
维护状态 低维护
是否开源 是
定价模式 free

项目数据

分类开源模型
开发团队open-thought
所属国家
官网地址
定价模式free
价格说明开源项目,Apache-2.0许可证,完全免费,可自行部署使用。
访问状态
是否开源是
开源协议Apache-2.0
主要语言Python
技术栈/模型gym,large-language-models,reinforcement-learning
GitHub 星标★ 1512
30天Star增速
HF 下载量
上线时间2025-01-23 00:00:00
最近更新2026-09-23 00:00:00
维护状态低维护
中文支持
访问方式
移动端支持
综合评分
收录时间2026-08-09
浏览次数4

使用教程

难度:入门 约 10 分钟 部署方式:库/依赖 6 步

环境要求

  • Python >= 3.10
  • 已安装 pip(用于从 PyPI 安装包)
  • 可访问 PyPI 网络(pip 下载)

安装与启动步骤

  1. 1确认 Python 版本

    reasoning-gym 要求 Python >= 3.10,先检查本机版本,低于 3.10 需先升级 Python。

    python --version
  2. 2安装 reasoning-gym

    从 PyPI 安装最新发布包,这是 README 给出的唯一安装命令。安装后即可在代码中 import。

    pip install reasoning-gym
  3. 3生成第一个数据集

    用 create_dataset 生成任务,指定任务名、样本数量与随机种子;再用 score_answer 做算法校验。

    python -c "import reasoning_gym
    data = reasoning_gym.create_dataset('leg_counting', size=10, seed=42)
    for i, x in enumerate(data):
        print(f'{i}: q="{x["question"]}", a="{x["answer"]}"')
        assert data.score_answer(answer=x['answer'], entry=x) == 1.0"
  4. 4传任务专属参数

    不同任务有各自的配置项,以关键字参数形式传入,例如 leg_counting 支持 max_animals。

    python -c "import reasoning_gym
    reasoning_gym.create_dataset('leg_counting', size=10, seed=42, max_animals=20)"
  5. 5组合多个任务

    用 DatasetSpec 构造 composite 数据集,weight 控制各任务占比,config 传各任务配置。

    python -c "import reasoning_gym
    from reasoning_gym.composite import DatasetSpec
    specs = [
        DatasetSpec(name='leg_counting', weight=2, config={}),
        DatasetSpec(name='figlet_font', weight=1, config={'min_word_len': 4, 'max_word_len': 6}),
    ]
    reasoning_gym.create_dataset('composite', size=10, seed=42, datasets=specs)"
  6. 6安装符号评分扩展

    需要 symbolic math 校验(cascade scorer)时,安装 scoring 附加依赖,减少格式差异造成的误判。

    pip install reasoning-gym[scoring]

关键配置

配置项必填说明示例
size是生成的任务样本条数10
seed是随机种子,保证数据集可复现42
max_animals否leg_counting 任务专属参数,限制动物数量20
datasets否composite 数据集的任务列表,元素为 DatasetSpec[DatasetSpec(name='leg_counting', weight=2, config={})]
DatasetSpec.weight否组合数据集中该任务的相对权重2
DatasetSpec.config否该任务的专属配置字典,空字典为默认配置{"min_word_len": 4, "max_word_len": 6}

如何确认成功

代码能 import reasoning_gym,create_dataset 正常打印题目与答案,且 score_answer(...) == 1.0 断言通过。

常见问题

Q:安装后 import 报错或版本不符?

A:确认 Python 版本 >= 3.10;README 说明 PyPI 版本可能比 main 分支落后几天,属正常现象。

Q:怎么校验模型输出的答案是否正确?

A:使用数据集自带的 score_answer 方法,对答案做算法化验证,返回 1.0 表示正确。

Q:答案格式略有差异(LaTeX、大小写)会判错吗?

A:可安装 reasoning-gym[scoring] 并使用 cascade_score,它依次尝试字符串、数值、符号数学匹配以降低误判。

Q:怎么开始用 RG 数据做 RL 训练?

A:README 推荐使用 verifiers 库(见 examples/verifiers),RG 数据也可接入任何主流 RL 训练框架。

注意事项

  • PyPI 发布版本可能落后 main 分支数天,追求最新功能需自行从源码安装。
  • scoring 附加依赖仅在需要符号数学验证时才需要安装。
  • 任务数量众多,除 leg_counting、figlet_font 外的任务名与参数请查阅官方 GALLERY.md 文档。

核心亮点

  • 内置40+种推理任务,覆盖算法、数学、逻辑,开箱即用
  • 每个任务都有可验证的奖励函数,无需人工标注,训练更高效
  • 接口统一,易于集成到现有RL框架,且支持自定义任务扩展

不足之处

  • 任务难度和多样性可能不足以覆盖所有复杂推理场景
  • 文档/社区待观察

适用场景

  • 训练LLM进行数学解题、逻辑推理等强化学习实验
  • 评估或比较不同RL算法在可验证推理任务上的表现
  • 为学术研究提供标准化的推理环境基准

替代项目

OpenAI Gym、BabyAI、TextWorld

项目介绍

reasoning-gym 是模型训练领域的开源项目,由 open-thought 开发,2025 年首次发布。

在全站 13,109 个收录项目中,它的 GitHub 星标数(1,512)位列前 30%,在模型训练分类中处于中上游。

近 42 天,它的 GitHub 星标从 1,481 增加到 1,512,净增 31。

项目已超过三个月没有代码更新,维护节奏明显放缓,最近一次代码更新于 2026-09-23。Apache-2.0许可证,完全免费,可自行部署使用。

它主要面向的使用场景是:训练LLM进行数学解题、逻辑推理等强化学习实验。同类可对比的替代方案包括 OpenAI Gym、BabyAI、TextWorld。

上一篇:flame

下一篇:Time-R1

同类项目推荐

Long-RL 开源

让强化学习轻松驾驭超长序列,训练更稳更快

Long-RL: Scaling RL to Long Sequences (NeurIPS 2025)

★ 729 2026-08-09
TPA 开源

把注意力复杂度从平方降到线性,长序列不再卡顿

[NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (https://arxiv.or···

★ 463 2026-08-09
minimind 开源

两小时造出你的小模型,练手入门不求人

Train a 64M-parameter LLM from scratch in just 2h!

★ 62178 2026-08-09
pruna 开源

一键优化 AI 模型,推理提速又省资源。

Pruna is a model optimization framework built for developers, enabling you to delive···

★ 1286 2026-08-13