inclusionAI/AReno
An easy-to-use, fast toolkit to scale up RL post-training on a single node.
解决的问题
AReno 是一套自包含的工具组,旨在让大型语言模型 (LLM) 的强化学习 (RL) 与后训练能在单个节点上轻松完成。它消除了对复杂集群设置或手动串接不同训练框架、推理服务器与内核库的需求,让开发者能从基础检查点 (checkpoint) 直接在本地完成模型部署。
运作方式
AReno 提供全栈设计,将训练、推理与 rollout 能力整合在单个套件中。它使用 Trainer 类来管理 RL 循环:生成策略内 (on-policy) 的完成内容 (rollout)、通过用户定义的奖励函数进行评分,并更新模型权重 (训练)。它支持多种硬件后端,包括 NVIDIA CUDA、Ascend NPU 与 Apple Silicon (通过 MLX),并提供与 OpenAI 兼容的代理程序 RL 轨迹代理服务器。
目标对象
希望进行快速、本地化 LLM 后训练、SFT/DPO 风格训练或代理程序 RL,且不想负担管理分布式集群开销的研究人员与开发者。
特色亮点
- 即插即用:通过简单的 CLI 标志或 Python API 访问各种后训练方法。
- 多后端支持:原生兼容 CUDA、Ascend NPU 与 Apple Silicon MLX。
- 代理程序 RL 就绪:支持从代理程序与本地 OpenAI 兼容代理服务器交互所产生的轨迹进行训练。
- 多模态能力:使用 OpenAI 风格的消息格式处理图像、音频与视频内容。
- LORA 集成:原生支持为 Qwen3 与 Bailing-MoE V3 等模型训练 TP 感知适配器,并保存标准 PEFT 产出物。
- 可扩展架构:允许轻松注册新的算法、奖励函数与硬件后端。
相关
- 项目
- 项目
- 项目
- 项目
- 项目