What problem does it solve?
Provides practical guidance for post-training large language models with reinforcement learning (RL), coordinating Megatron-LM training and SGLang-based high-throughput rollouts so you can scale RL fine-tuning for GLM-family and related models.
Core Features & Use Cases
- Megatron-LM RL Training: Runs actor/critic training with tensor/pipeline/data parallelism for scalable optimization.
- SGLang Rollout Generation with Routing: Generates rollouts efficiently through SGLang routers to support high-throughput reward collection.
- Flexible Data Buffer & Workflows: Manages prompts and generated samples with off-policy buffering and custom generation functions for multi-turn or tool-use settings.
- Algorithms & Config Options: Supports GRPO-style group-relative advantage training, KL loss options, asynchronous training, and evaluation across multiple tasks.
Quick Start
Ask the AI to walk you through running GRPO training by sourcing a model script, selecting rollout and batch parameters, and launching train.py with your JSONL prompt dataset and label keys.