NVIDIA发布Molt:专为智能体强化学习优化的PyTorch原生框架

NVIDIA AI Releases Molt: A PyTorch-Native Agentic Reinforcement Learning Framework

NVIDIA近日在AI领域动作频频,继推出高性能计算芯片后,其软件生态建设也迈出了关键一步。针对智能体强化学习这一前沿领域,NVIDIA正式开源了一款名为“Molt”的新框架。这一框架旨在解决当前主流强化学习研究中面临的高昂开发成本问题,为研究人员提供了一个更加轻量、高效且原生的开发平台。

在当前的智能体强化学习研究中,算法的迭代速度往往受限于底层框架的复杂度。传统的强化学习框架通常将训练器、分布式后端和 rollout(策略执行)逻辑紧密耦合。这意味着,研究人员每提出一个新的算法想法,往往都需要在庞大的代码库中进行复杂的修改,这无疑大大增加了研发周期和出错的风险。

Molt 框架的出现正是为了打破这一僵局。NVIDIA 团队透露,Molt 仅包含约 8.6 万行代码,相比许多臃肿的传统框架,其代码量极其精简。该框架巧妙地将 Ray(分布式计算)、vLLM(高性能推理)和 NeMo AutoModel(模型管理)等组件,通过一个核心的异步循环进行了有机整合。

作为一款 PyTorch 原生框架,Molt 的设计理念充分尊重了 Python 开发的灵活性。它允许智能体逻辑保持“普通 Python”风格,无需为了适应框架而进行繁琐的底层重构。同时,Molt 保证了轨迹的 Token 精确性,这对于强化学习中的策略评估至关重要。在性能测试中,Molt 展现出了令人惊讶的吞吐量,其统计表现与基于 Megatron 的高性能堆栈相当。

这一突破对于整个行业具有深远意义。随着大模型向智能体方向发展,能够快速迭代、性能强大的强化学习工具将成为核心竞争力。Molt 的出现,不仅降低了智能体算法开发的门槛,也为后续的自动化研究提供了坚实的底座。对于开发者而言,这意味着他们可以将更多精力集中在算法创新本身,而非被底层基础设施所牵绊。

信息来源:MarkTechPost,原文链接:https://www.marktechpost.com/2026/08/01/nvidia-ai-releases-molt-a-pytorch-native-agentic-reinforcement-learning-framework/

封面图片来源:Unsplash / 摄影师 Brecht Corbeel

oaido_ai

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注