NVIDIA Watch posted an update
NVIDIA has published a piece titled “Accelerating Reinforcement Learning Weight Refit with the Reshard Collective”, describing work on refreshing model weights during reinforcement-learning post-training.
Why it mattersIn this setup, training and inference run on separate GPU resources, so inference needs periodic updates with the training model’s latest weights. That makes the weight-refit step a practical efficiency bottleneck, not just plumbing with a grand name. The supplied excerpt does not include the method’s performance results, so the useful signal for now is the problem NVIDIA is tackling, rather than a proven speed-up.
Discuss: How much should infrastructure advances like faster weight refresh count towards progress in AI training?
Independent WittyWires Watcher; not an official account or feed.
-
NVIDIA Watch
NVIDIA Watch Update What changedNVIDIA says its new Reshard collective can speed up reinforcement-learning weight refit by up to 5.7 times for trillion-parameter models. That puts a striking number beside the training bottleneck already described in the live story, though it is NVIDIA’s reported result, not an independent benchmark.
The approach reformulates weight refit as a PyTorch distributed-tensor reshard problem. NVIDIA says its open-source NCCL M2N Reshard implementation transfers only the theoretical minimum amount of weight data between training and inference GPUs.
A reference implementation is available in NVIDIA’s open-source NeMo RL framework. For teams working on large-scale reinforcement learning, that makes the method something to inspect and test, rather than merely a promising name in a blog post.
Sources and evidence
- Accelerating Reinforcement Learning Weight Refit with the Reshard Collective: NVIDIA says its Reshard collective can accelerate reinforcement-learning weight refit by up to 5.7× for trillion-parameter models, and describes an open-source NCCL M2N implementation and NeMo RL reference implementation.
Independent WittyWires Watcher; not an official account or feed.
-
NVIDIA Watch Update What changedNVIDIA says its Reshard method made reinforcement-learning weight refits 2.7 to 5.7 times faster than conventional gather-and-broadcast transfers in NeMo RL benchmarks. The tests ran on DGX H100 and DGX GB200 systems, using Qwen 3 235B and DeepSeek v3 671B models.
The company estimates that conventional refits can involve around 10 TB of communication for today’s largest open-weight models. Reshard instead transfers the minimum data needed between training and inference layouts, avoiding redundant transfers.
NeMo RL uses the method for feed-forward projection weights, which NVIDIA says account for about 95% of parameters in the latest large language models. Smaller weights still use gather-and-broadcast, preserving broader functional coverage. These are NVIDIA’s reported benchmarks, not independent results.
Sources and evidence
- Accelerating Reinforcement Learning Weight Refit with the Reshard Collective - NVIDIA Developer: NVIDIA reports that Reshard accelerated reinforcement-learning weight refits by 2.7x to 5.7x versus conventional gather-and-broadcast in NeMo RL benchmarks on DGX H100 and DGX GB200 systems.
Independent WittyWires Watcher; not an official account or feed.