行业精选

Cognition披露生产级RL训练基础设施:跨三大洲四数据中心

This is what production RL infra looks like. ICYMI: RL training at scale separates into two distinc...

精选理由

Cognition分享了他们怎么跨三大洲四个数据中心做RL训练,用紧耦合trainer加分布式异步推理,还靠压缩权重差异同步,做到稳定rollout。做大规模RL的值得看看。

AI 摘要

Cognition团队分享了其生产级强化学习训练基础设施的设计,核心将训练分为两部分:紧耦合通信的trainer和分布式异步rollout推理。他们的RL训练横跨三大洲四个数据中心,结合自有GPU集群和Fireworks AI等推理提供商的算力。rollout推理引擎通过压缩权重差异在对象存储中同步,实现分布式扩展。这套基础设施支撑了SWE-1.7模型的训练,并确保跨数据中心的可靠rollout。

AI 翻译 · 中文

Cognition团队分享了其生产级强化学习训练基础设施的设计,核心将训练分为两部分:紧耦合通信的trainer和分布式异步rollout推理。他们的RL训练横跨三大洲四个数据中心,结合自有GPU集群和Fireworks AI等推理提供商的算力。rollout推理引擎通过压缩权重差异在对象存储中同步,实现分布式扩展。这套基础设施支撑了SWE-1.7模型的训练,并确保跨数据中心的可靠rollout。

Fireworks AIThis is what production RL infra looks like. ICYMI: RL training at scale separates into two distinct problems. - Tight collective comms for the trainer. - Distributed async inference for rollout. Kudos to the @cognition