What problem does it solve?
This Skill provides a comprehensive framework for large language model (LLM) post-training with reinforcement learning (RL) using the slime tool, addressing challenges such as RL scaling and custom data generation workflows.
Core Features & Use Cases
- Megatron-LM Integration: Offers seamless integration with Megatron-LM for training.
- Custom Data Generation: Supports custom data generation workflows for flexible prompt management and sample storage.
- High-Throughput Rollout Generation: Leverages SGLang for high-throughput rollout generation.
- Model Support: Supports training of GLM, Qwen3, DeepSeek V3/R1, and Llama 3 models.
- Use Case: Ideal for research-grade framework users looking to implement RL scaling for large models with tight integration of Megatron-LM and SGLang.
Quick Start
Train a GLM model using slime by executing the following command in your terminal:
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