Prime Intellect 发布 prime-rl 0.6.0,用于训练万亿参数 MoE 模型的智能体强化学习任务
Prime Intellect 发布了 prime-rl 0.6.0,这是一个用于万亿参数混合专家(MoE)模型异步强化学习的开源框架。该框架在 SWE 任务上训练了 GLM-5,序列长度高达 131k,步时低于 5 分钟,使用 256 次 rollout 和 28 个 H200 节点。本文详细介绍了其推理和训练优化,包括 FP8 推理、宽专家并行、预填充/解码分离、路由器重放以及 3D 并行(FSDP、EP、CP)。
Prime Intellect 发布 prime-rl 0.6.0,用于训练万亿参数 MoE 模型的智能体强化学习任务
本文仅提供摘要
根据来源内容权限,本站仅提供编辑摘要、要点和原文链接。