What problem does it solve?
LLM developers struggle to efficiently run reinforcement learning post-training when training needs Megatron-LM parallelism while rollouts require high-throughput generation, causing slow iteration and expensive GPU usage.
Core Features & Use Cases
- Megatron-LM Training Loop: Actor (and optional critic) training with full TP/PP/DP/SP parallelism for large-scale model updates.
- SGLang Rollouts with Router: High-throughput response generation using SGLang, optionally routed for better utilization and multi-turn rollout support.
- Flexible Data Buffering (Off-Policy Ready): Prompt management plus sample storage and optional buffer filtering to support GRPO/GPSO/PPO-style workflows and custom data generation.
Use case example: You want to post-train a GLM-family model using GRPO on reasoning prompts, generating many rollouts per prompt and feeding rewards back into Megatron-LM training with tight rollout/training integration.
Quick Start
Ask your AI to set up GRPO training for your model by providing the model script to source, your JSONL prompt/label data paths, and the desired batch and rollout parameters so it can launch train.py with the correct keys and estimation settings.