PrimeIntellect-ai/prime-rl

Agentic RL Training at Scale

What it solves

prime-rl addresses the challenge of scaling reinforcement learning (RL) for large-scale agentic training. It provides a framework that allows developers to train massive models (up to 1T+ parameters) across thousands of GPUs while maintaining a modular and "hackable" design for rapid experimentation.

How it works

The framework utilizes a fully asynchronous RL architecture to maximize throughput. For training and inference at scale, it integrates several high-performance technologies:

  • Training & Inference: Uses FSDP2 for training and vLLM for inference, incorporating FP8 precision, Expert Parallelism (EP), and Context Parallelism (CP).
  • Deployment: Supports multi-node deployment via Slurm, including advanced configurations like P/D disaggregation and KV offloading.
  • Pipeline: Provides an end-to-end post-training pipeline covering Supervised Fine-Tuning (SFT), RL training, and evaluations.
  • Integration: Connects natively with the Environments Hub and verifiers for agentic and software engineering (SWE) environments.

Who it’s for

It is designed for AI researchers and engineers working on frontier models, specifically those needing to perform large-scale RL post-training on MoE (Mixture-of-Experts) or dense models, and those training VLMs (Vision-Language Models).

Highlights

  • Massive Scalability: Capable of scaling to 1000+ GPUs for training 1T+ MoE models.
  • Broad Model Support: Native support for GLM, Qwen, Nemotron, and other major MoE families.
  • Multimodal Capability: Supports training for VLMs such as Qwen3.5.
  • Flexible Deployment: One-line Slurm deployment for complex frontier model setups.
  • Comprehensive Tooling: Includes built-in support for SFT, RL, and evaluation in a single framework.

Related

  • Project
  • Project
  • Project
  • Project
  • Dispatch