What problem does it solve?
This Skill addresses the challenge of training large language models (LLMs) with reinforcement learning (RL) at scale, offering a flexible infrastructure backend for RLHF, GRPO, PPO, and other RL algorithms.
Core Features & Use Cases
- Scalable RL Training: Supports RL training for LLMs up to 671B parameters.
- Flexible Backends: Allows swapping backends like FSDP, Megatron-LM, and vLLM.
- Multiple RL Algorithms: Includes support for PPO, GRPO, RLOO, REINFORCE++, and DAPO.
- Multi-Turn Rollout: Enables multi-turn rollout with tool calling for agentic workflows.
- Vision-Language Models: Supports vision-language model RL training.
- Use Case: Ideal for researchers and engineers looking to implement distributed RL training for LLMs, with a focus on math reasoning and reasoning models.
Quick Start
Train a reasoning model on math tasks like GSM8K or MATH using verl with the following command:
python3 -m verl.trainer.main_ppo \
algorithm.adv_estimator=grpo \
data.train_files=~/data/gsm8k/train.parquet \
actor_rollout_ref.model.path=Qwen/Qwen2.5-7B \
actor_rollout_ref.rollout.n=8 \
actor_rollout_ref.actor.use_kl_loss=True \
trainer.n_gpus_per_node=8