What problem does it solve?
Reduces the complexity of performing reinforcement-learning-based post-training for large language models by integrating Megatron-LM training with high-throughput SGLang rollouts, flexible data buffering, and customizable reward/generation logic so teams can move from research experiments to scalable training runs.
Core Features & Use Cases
- Megatron-LM native training with full parallelism (tensor, pipeline, data, sequence) to train large GLM, Qwen3, DeepSeek, and Llama 3 models.
- SGLang rollouts and router for high-throughput generation, multi-turn agent workflows, and tool-calling scenarios.
- Flexible data buffer and off-policy reuse enabling async training, buffered rollouts, prioritized sampling, and GRPO-style advantage estimators.
- Custom hooks for generate functions and reward models to support multi-turn tool-enabled agents and bespoke evaluation metrics.
- Operational guidance for Docker/source installs, async vs synchronous modes, colocated inference, and common troubleshooting for OOMs, weight sync, and router imbalance.
Quick Start
Run a GRPO post-training run that uses Megatron-LM for actor updates and SGLang for rollouts on your GLM checkpoint with a buffered prompt dataset.