What problem does it solve?
This Skill addresses the complexity and scalability challenges of post-training reinforcement learning (RL) for large language models (LLMs), specifically targeting the tight integration of Megatron-LM and SGLang frameworks.
Core Features & Use Cases
- Megatron-LM Training: Full parallelism support for actor model training with Megatron-LM.
- Rollout Generation: High-throughput rollout generation using SGLang-based router.
- Data Buffer: Flexible prompt management and sample storage for custom data generation workflows.
- Model Support: Compatibility with GLM, Qwen3, DeepSeek V3/R1, Llama 3, and other large models.
- Use Case: Ideal for research-grade frameworks seeking production backing, especially for custom data generation workflows and tight integration with Megatron-LM.
Quick Start
Run the following command to begin a standard GRPO training session with the provided model and data:
python train.py --actor-num-nodes 1 --actor-num-gpus-per-node 4 --rollout-num-gpus 4 --advantage-estimator grpo --use-kl-loss --kl-loss-coef 0.001 --rollout-batch-size 32 --n-samples-per-prompt 8 --global-batch-size 256 --num-rollout 3000 --prompt-data /path/to/data.jsonl