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

NVIDIA's NeMo team released Molt, a PyTorch-native agentic reinforcement learning framework designed to be compact enough for researchers and AI coding assistants to grasp, with roughly 8.6K lines of RL code compared to about 62K for verl, 25K for slime, and 7.2K for OpenRLHF. Molt ships under Apache 2.0 with launch codes, Slurm scripts, and a prebuilt container, but requires 2 nodes of 8 H100 GPUs for the shipped recipes. The framework composes Ray, vLLM, and NVIDIA AutoModel without forking, and enforces correctness invariants including rollout routing replay for mixture-of-experts policies.

Agentic reinforcement learning research is constant algorithm modification. New estimators, new pipeline stages, new rollout schemes. In mainstream frameworks each change threads through layers of trainer, distributed backend, and rollout glue. That cost lands on the researcher at every iteration. Molt https://github.com/NVIDIA-NeMo/labs-molt , from NVIDIA’s NeMo team, targets that cost directly. Its a PyTorch-native agentic RL framework with an unusual design target. The codebase should be compact enough for a researcher to hold in their head, and for an AI coding assistant to read and reason about in its entirety. The stated footprint is roughly 8.6K lines of RL code, measured by tracing the import graph from each framework’s RL entry point. The same method counts about 62K lines for verl, 25K for slime, and 7.2K for OpenRLHF. Is it deployable? Yes. Molt ships under Apache 2.0 with launch codes, Slurm scripts, and a prebuilt container. But the research paper positions it as research infrastructure, not a production training service, and hardware is the real gate. The shipped recipes assume 2 nodes of 8 H100 GPUs, split 8 for training and 8 for rollout. That puts Molt in reach of frontier and frontier-adjacent labs, well-funded AI startups doing post-training, enterprise AI research groups in finance, healthcare, and robotics that train agents against proprietary environments, and academic labs with multi-node H100/H200 access. Applications include multi-turn tool-use agents, code-execution agents, vision-language environments the shipped geo3k recipe , LLM-as-judge reward loops, and on-policy distillation onto a smaller student. Three components, one loop Molt composes Ray https://github.com/ray-project/ray for placement and asynchronous queues, vLLM https://github.com/vllm-project/vllm for rollout, and NVIDIA AutoModel with FSDP2 for training. None of the three is forked, so upstream improvements arrive as a container pin rather than a rebase. The runtime is an agent pool, a set of vLLM engines behind a request router, and a single trainable policy actor. A streaming pool keeps prompt groups in flight so engines never drain while the actor trains. Partial rollout pauses the engines, broadcasts actor shards over NCCL directly to each engine, and resumes retained requests instead of discarding them. The agent is an ordinary program An RL run names one Python module that exports an AgentRunner . Everything else is ordinary code, including the reward. Two forms are supported. With Env , the framework owns the LLM loop in a Gymnasium-aligned step . With ChatAgent , the user owns the loop through a stock OpenAI or Anthropic SDK. Molt launches a loopback server that speaks both wire protocols, and every request decodes server-side into one token-exact accumulation. When a long-horizon agent compacts its context and rewrites the prefix, the server seals the current segment and opens a fresh one automatically. Never train on a token you did not generate Three correctness invariants organize the design. Token identity: sampled token ids define the trajectory, not a retokenized transcript. Policy-version semantics: trainable tokens keep their behavior-policy log-probabilities, and asynchronous use is corrected per token behind a sequence-level gate. Forward consistency: rollout and actor must agree on model semantics. For mixture-of-experts policies, the last invariant matters most. The rollout and training routers select experts independently, and small numerical differences can flip top-k choices. Molt applies rollout routing replay https://arxiv.org/abs/2510.11370 , where vLLM returns its per-token expert ids and the training forward replays them. Interactive explainer Key Takeaways - Molt is an Apache-2.0 agentic RL framework in about 8.6K lines of RL code, roughly 7× smaller than verl. - Ray, vLLM, and NVIDIA AutoModel are composed, never forked, so upstream releases arrive as a container pin. - Agents are plain Python; a stock OpenAI or Anthropic SDK trains as-is through a token-exact loopback server. - Throughput is statistically comparable to a Megatron-based stack, with the MoE-mismatch caveat disclosed. - Scale is a flag: the same loop runs a dense 4B model and a 700B MoE at --fsdp.ep size 256 . Check out the Paper and Also, feel free to follow us on Repo here https://github.com/NVIDIA-NeMo/labs-molt . and don’t forget to join our Twitter https://x.com/intent/follow?screen name=marktechpost and Subscribe to 150k+ML SubReddit https://www.reddit.com/r/machinelearningnews/ . Wait are you on telegram? our Newsletter https://www.aidevsignals.com/ now you can join us on telegram as well. https://t.me/machinelearningresearchnews Need to partner with us for promoting your GitHub Repo OR Hugging Face Page OR Product Release OR Webinar etc.? Connect with us https://forms.gle/wbash1wF6efRj8G58 Asif Razzaq is the CEO of Marktechpost Media Inc.. As a visionary entrepreneur and engineer, Asif is committed to harnessing the potential of Artificial Intelligence for social good. His most recent endeavor is the launch of an Artificial Intelligence Media Platform, Marktechpost, which stands out for its in-depth coverage of machine learning and deep learning news that is both technically sound and easily understandable by a wide audience. The platform boasts of over 2 million monthly views, illustrating its popularity among audiences.