NVIDIA AI Releases Molt: A PyTorch-Native Agentic Reinforcement Learning Framework
MarkTechPost Asif Razzaq
NVIDIA's NeMo team released Molt, a lean PyTorch-native framework for training AI agents with reinforcement learning. It's roughly 7x smaller than rival frameworks, but still needs serious GPU firepower to run.
Based on reporting by MarkTechPost, Asif Razzaq — read the original for the full story.
Summary, retelling and take written by AI under human oversight; images are AI-generated illustrations. How we work · Report an error
Agentic reinforcement learning research is basically an endless cycle of tweaking algorithms — new estimators, new pipeline stages, new ways of handling rollouts. The problem is that in most existing frameworks, every one of those tweaks has to thread through layers of trainer code, distributed backend logic, and rollout plumbing. Researchers pay that tax on every single iteration. NVIDIA's NeMo team built Molt specifically to cut that tax down.
The pitch is simplicity as a feature. Molt's RL code comes in at about 8,600 lines, measured by tracing the import graph from the framework's entry point. Compare that to verl at roughly 62,000 lines, slime at 25,000, and OpenRLHF at 7,200. NVIDIA's goal was a codebase small enough for a human researcher to hold in their head — and small enough for an AI coding assistant to read start to finish and actually reason about, rather than just autocomplete around the edges.
Under the hood, Molt stitches together three existing tools rather than reinventing them: Ray handles placement and async queues, vLLM runs the rollouts, and NVIDIA AutoModel with FSDP2 handles training. Crucially, none of these get forked, so when upstream ships an improvement, Molt just pins a new container version instead of undergoing a painful rebase. The runtime itself is a pool of vLLM engines sitting behind a request router, feeding a single trainable policy actor, with a streaming pool that keeps prompts flowing so the engines never sit idle while training happens.
Writing an agent for Molt is refreshingly boring, in a good way. You just point the framework at a Python module exporting an AgentRunner, and everything else — including the reward function — is ordinary code. You can let Molt own the LLM loop through a Gymnasium-style step() call, or keep control yourself using a standard OpenAI or Anthropic SDK, which Molt intercepts through a loopback server that decodes every request into an exact token accumulation. That detail matters a lot for correctness: three invariants govern the whole design, ensuring the framework trains only on tokens the model actually generated, preserves the right log-probabilities for async rollouts, and keeps rollout and training in agreement on model behavior — a particularly thorny issue for mixture-of-experts models, where Molt replays vLLM's expert-routing choices during training to avoid mismatches.
None of this is free, though. The shipped recipes assume two nodes of eight H100 GPUs apiece, split evenly between training and rollout. That puts Molt squarely in the hands of frontier labs, well-capitalized startups doing post-training, enterprise research teams in finance or healthcare or robotics, and academic groups lucky enough to have multi-node H100 or H200 clusters. It's released under Apache 2.0 with Slurm scripts and a prebuilt container, and NVIDIA is careful to frame it as research infrastructure rather than a production training service — a distinction that matters once you look at the hardware bill.
My take — AI-written commentary, not fact-checked reporting
Shrinking a research codebase from 62,000 lines to 8,600 is a genuinely useful contribution, and I like that NVIDIA didn't fork Ray, vLLM, or AutoModel to get there — that's the difference between a framework that ages well and one that rots the moment upstream moves on. But let's not pretend 'Apache 2.0 and open' means 'accessible': needing 16 H100s just to run the reference recipes keeps this squarely in big-lab territory, which is the same story open-weights AI keeps telling us — the license is free, the compute isn't.
Read more about this at: MarkTechPost
Related stories
NVIDIA AI Releases Nemotron 3.5 Lightning: A 30B Open MoE with 3B Active Parameters, and NeMo Switchyard Model Router
MarkTechPost · 1 month ago ·
10
Nvidia releases Nemotron 3.5 Lightning and NeMo Switchyard to give enterprise AI capability options
SiliconANGLE · 1 month ago ·
31