LlamaRL: A Distributed Asynchronous Reinforcement Learning Framework for Efficient Large-scale LLM Training
Meta 开源的全分布式异步强化学习框架,基于原生 PyTorch 实现单控制器架构,支持 8B 到 405B 参数模型在数千 GPU 上的 RLHF/RL 训练。提供了异步设计严格加速的理论证明。
INSIGHTLlamaRL 是 Llama 3 后训练中实际使用的 RL 框架,其单控制器 + 异步设计在工程上极具参考价值;对搜索排序模型的 RLHF 训练也可直接复用。
业务关联搜索排序模型越来越多地采用 RLHF/RLEF 进行对齐训练,LlamaRL 的分布式异步架构可显著提升 RL 训练效率,减少 GPU 空闲等待。
证据在 Llama 3 后训练中实际应用,支持 8B/70B/405B 模型规模;理论证明异步设计带来严格加速;基于原生 PyTorch 可扩展到数千 GPU。