ZH ▾
SNAPE REP RT 头版 →
已证实算力11月06日 18:03 UTC

英伟达开源 Reshard 集合通信,强化学习权重 refit 提速最高 5.7 倍

[object Object]
NVIDIA Technical Blog

NCCL M2N Reshard 只传输相交分片;NeMo RL 基准相对 gather-and-broadcast 提速 2.7–5.7 倍。

在 NVIDIA Technical Blog 阅读全文 分享

为何重要

英伟达介绍 Reshard:将强化学习权重 refit 视为 DTensor 到 DTensor 传输,只搬迁理论最小相交分片。

要点

  • New Reshard collective for DTensor-to-DTensor refit
  • Open NCCL M2N Reshard reference implementation
  • 2.7×–5.7× faster than gather-and-broadcast in NeMo RL
  • Benchmarked on DGX H100 and GB200
  • Hybrid path: FFN projections via Reshard; small weights legacy
  • Aimed at trillion-parameter RL post-training payloads

来源

更多算力报道
Snape Report 更多报道头版 →