What problem does it solve?
This Skill helps you run LLM post-training using reinforcement learning at scale by connecting Megatron-LM training with SGLang rollout generation, so you can iterate on RL objectives and data pipelines without manual glue code.
Core Features & Use Cases
- Megatron-LM RL Training: Enables actor/critic style post-training with parallelism (TP/PP/DP/SP) while keeping training tightly integrated with rollout weights.
- SGLang High-Throughput Rollouts: Uses a router-backed SGLang engine to generate many responses efficiently and feed reward/verifier outputs back into training.
- Flexible Data Buffer & Multi-Turn Generation: Supports custom data generation, filtering (off-policy buffering), and multi-turn/tool-calling workflows via a custom generate function.
Use case: you want to train a GLM/Qwen/DeepSeek/Llama reasoning model with GRPO/GPSO/PPO-style objectives, generating multiple candidate responses per prompt and selecting or weighting them using rewards.
Quick Start
Start GRPO training by running the repository’s GRPO example command configured for your model and dataset, and point it to your JSONL prompts and labels.