THUDM/slime
slime is an LLM post-training framework for RL Scaling.
它解决了什么问题
slime 是一个专为 RL 扩展设计的高性能 LLM 后训练框架。它通过将训练、Rollout 和数据生成路径统一为单一、简化的流程,解决了训练器、Rollout 服务和 Agent 框架之间脱节且沉重的技术堆栈问题。
运作原理
slime 连接了用于训练的 Megatron-LM 和用于 Rollout 的 SGLang。它采用“原生直通”设计,允许用户直接使用 Megatron 和 SGLang 的参数,而无需增加额外的抽象层。架构包含三个主要模块:
- 训练 (Megatron):处理主要的训练过程并将参数同步到 Rollout 模块。
- Rollout (SGLang + router):生成新数据并计算奖励/验证器输出。
- 数据缓冲区 (Data Buffer):作为管理提示词初始化和 Rollout 生成方法的桥梁。
它支持多种进阶部署配置,例如 PD 分离(预填充/解码分离)、用于高效更新的 Delta 权重同步,以及用于分离服务的外部 Rollout 引擎。
适用对象
致力于 LLM 大规模 RL 后训练的研究人员与工程师,特别是那些使用 Megatron-SGLang 技术堆栈的团队。它专为需要一个轻量级且经过实战验证的框架,并希望将其扩展到 Agent RL、全模态训练或可验证环境的用户而设计。
特点
- 实战验证:用于训练 GLM 系列模型(从 GLM-4.5 到 GLM-5.3)。
- SGLang 原生:针对 SGLang 进行了深度优化,以利用其特定的服务、路由和缓存功能。
- 灵活的数据生成:支持多 Agent 系统、工具使用和长程 Agent 工作流程的自定义接口,无需 Fork 训练核心。
- 生产就绪:内置对容错、追踪、性能分析以及完整的 RL 基础设施 CI/CD 的支持。
- 广泛的模型支持:兼容于 Qwen、DeepSeek V3/R1 和 Llama 3 系列。
相关
- 项目
- Dispatch
- Dispatch
- 项目
- 项目