NVIDIA NeMo Open-Sources Molt, a Lean PyTorch Agentic RL Stack
TL;DR
- NVIDIA NeMo released Molt, a PyTorch-native training framework designed to be readable end to end by both researchers and coding assistants.
- The paper claims Molt is statistically comparable to a state-of-the-art Megatron-based stack under a matched fully-asynchronous protocol.
- vLLM over Ray serves as Molt's rollout engine and, per the vLLM team, scales the setup to 1T-class MoE models.
NVIDIA's NeMo team has posted a paper that reframes what "framework" is supposed to mean in agentic RL. The pitch for Molt is deliberately anti-framework, a PyTorch-native training loop small enough for a researcher to hold in their head, and small enough for a coding assistant to read end to end.
The problem is familiar to anyone doing RL research on LLMs. Each new estimator, rollout scheme, or pipeline stage has to be threaded through stacked layers of trainer code, distributed backend, and rollout glue, and the cost of that friction lands on the researcher at every iteration. Molt's answer is to compress that stack into one asynchronous loop, treat the agent as "an ordinary program", and enforce a strict rule that the policy never trains on a token it did not itself generate. The authors claim that under a matched fully-asynchronous protocol, the lean version is "statistically comparable to a state-of-the-art Megatron-based stack", which is the interesting bit, leanness without an obvious capability tax.
The rollout piece comes from vLLM. The vLLM team confirmed the integration on X, saying vLLM over Ray "carries the rollout here: fast async serving up to 1T-class MoE scale, simple to drop in." That is the scale claim that would otherwise be easy to dismiss on a hackable stack.
Why this matters if you are on a small research team or in a university lab: the price of experimenting with agentic RL has largely been the framework tax. You either accept a heavy stack or you write your own trainer. A readable PyTorch codebase with a plausible Megatron-class performance ceiling shifts that tradeoff.
The honest caveat is that the abstract does not quantify how many lines of code Molt actually is, does not name the MoE model it benchmarks on, and does not detail the tasks behind the parity claim. Take the "statistically comparable" line as reported, not settled, until people outside NVIDIA reproduce the recipes in the labs-molt repo. The forward-looking piece is who moves first. Labs currently blocked on Megatron's learning curve are the ones with the most to gain.
Originally reported by paper
Read the original article →Original headline: NVIDIA NeMo Ships Molt: Hackable 9K-Line PyTorch Stack for Trillion-Scale Agentic RL