NVIDIA AI Releases Molt: A PyTorch-Native Agentic Reinforcement Learning Framework
NVIDIA
NVIDIA's NeMo team has open-sourced Molt, a compact PyTorch-native agentic reinforcement learning framework under Apache 2.0. With about 8.6K lines of RL code, it aims to reduce researcher overhead by composing Ray, vLLM, and NVIDIA AutoModel without forking, and supports correctness invariants like token identity and rollout routing replay. The shipped recipes assume 2 nodes of 8 H100 GPUs, targeting frontier and adjacent labs.
NVIDIA's NeMo team has released Molt, a PyTorch-native agentic reinforcement learning framework designed to reduce the research overhead of constantly modifying RL algorithms. The codebase is intentionally compact—roughly 8.6K lines of RL code, compared to about 62K for verl and 25K for slime—so researchers and AI coding assistants can grasp it entirely. Molt is deployable under Apache 2.0 with launch codes, Slurm scripts, and a prebuilt container, but the paper positions it as research infrastructure, not a production service. The shipped recipes assume 2 nodes of 8 H100 GPUs, split with 8 for training and 8 for rollout, putting it in reach of frontier labs, well-funded startups, enterprise research groups, and academic labs with multi-node H100/H200 access. Molt composes Ray for placement and queues, vLLM for rollout, and NVIDIA AutoModel with FSDP2 for training—none forked, so upstream improvements arrive as a container pin. The runtime uses an agent pool of vLLM engines behind a request router and a single trainable policy actor, with partial rollout pausing engines and broadcasting actor shards via NCCL. An RL run exports an AgentRunner module, supporting Env (Gymnasium-aligned step()) or ChatAgent (user-owned loop via OpenAI or Anthropic SDK) through a loopback server that ensures token-exact accumulation. Design correctness invariants include token identity, policy-version semantics, and forward consistency; for mixture-of-experts policies, Molt applies rollout routing replay where vLLM returns per-token expert ids and the training forward replays them. Throughput is statistically comparable to a Megatron-based stack, with the MoE-mismatch caveat disclosed, and the same loop runs a dense 4B model or a 700B MoE at --fsdp.ep_size 256.
- Сокращения
- PyTorch = Python-based scientific computing package
- RL = Reinforcement Learning — обучение с подкреплением
- GPU = Graphics Processing Unit — графический процессор
- MoE = Mixture of Experts — смесь экспертов
- FSDP = Fully Sharded Data Parallel — полностью шардированный параллелизм данных
- NCCL = NVIDIA Collective Communications Library — библиотека коллективных коммуникаций NVIDIA
- LLM = Large Language Model — большая языковая модель
- H100 = Hopper 100 (GPU architecture) — графический процессор архитектуры Hopper
Source: MarkTechPost —
original
