What problem does it solve?
This Skill provides guidance for training large language models (LLMs) using reinforcement learning (RL) with verl, addressing the need for scalable, flexible infrastructure backends and support for various RL algorithms.
Core Features & Use Cases
- Reinforcement Learning Training: Offers instructions for implementing RL algorithms like RLHF, GRPO, PPO, and others for LLM post-training.
- Scalable Infrastructure: Supports production-ready RL training at scale with flexible infrastructure backends.
- Algorithm Support: Provides support for multiple RL algorithms, including PPO, GRPO, RLOO, REINFORCE++, DAPO, and more.
- Use Case: Ideal for teams looking to scale RL training for LLMs, with examples like training reasoning models on math tasks like GSM8K or MATH.
Quick Start
Train a GRPO model for math reasoning using the verl skill:
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