What problem does it solve?
This Skill provides guidance for LLM post-training with Reinforcement Learning (RL) using slime, a Megatron+SGLang framework, enabling efficient training of GLM models and custom data generation workflows.
Core Features & Use Cases
- Megatron-LM Training: Offers full parallelism support for actor model training with Megatron-LM.
- SGLang Rollout: Utilizes SGLang for high-throughput rollout generation.
- Data Buffer: Provides flexible prompt management and sample storage.
- Model Support: Supports GLM, Qwen3, DeepSeek V3/R1, Llama 3 models.
- Use Case: Ideal for scaling RL in Megatron-LM with tight integration and custom data generation capabilities.
Quick Start
Run the following command to initiate a GRPO training session:
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