What problem does it solve?
LLM RL post-training is hard to scale and wire together correctly when you need efficient rollouts plus tightly integrated Megatron-LM training. This Skill helps you set up and run RL post-training using THUDM’s slime framework so that training and rollout generation work as a single system.
Core Features & Use Cases
- Megatron-LM training with full parallelism: Runs actor training (and optional critic) with tensor/pipeline/data parallelism to match large-model training setups.
- SGLang-based high-throughput rollouts: Generates rollout samples efficiently via SGLang and a router, enabling GRPO-style group-relative optimization loops.
- Flexible data buffer and custom generation: Supports off-policy buffering and custom multi-turn/tool-calling generation workflows for research-grade agents.
- Use Case: Train a GLM/Qwen/DeepSeek/Llama model with GRPO on JSONL prompts and labels, automatically generating multiple responses per prompt and optimizing using KL-regularized RL.
Quick Start
Use the slime skill to run GRPO training by executing a train.py command with your sourced model args, specifying prompt-data, input-key, label-key, rollout batch sizing, and rollout sample counts.