TensorHub: Scalable and Elastic Weight Transfer for LLM RL Training
cs.DC, cs.AI
Submitted: 2026-04-10
Updated: 2026-09-29
Journal ref: Proceedings of the ACM SIGOPS 32nd Symposium on Operating Systems Principles (SOSP '26), pp. 882-897, 2026
Code: https://github.com/NVIDIA/nccl
License: http://creativecommons.org/licenses/by/4.0/
The gist: Modern LLM reinforcement learning (RL) workloads require a high-performance weight transfer system to scale training across heterogeneous compute resources.
Terminology
Abstract
Modern LLM reinforcement learning (RL) workloads require a high-performance weight transfer system to scale training across heterogeneous compute resources. However, efficiently transferring terabyte-scale model weights across thousands of GPUs remains challenging because the system must accommodate clusters that dynamically scale up and down while keeping coordination, data movement, and storage overhead low. We introduce Reference-Oriented Storage (ROS), a new storage abstraction for RL weight transfer that exploits highly replicated model weights in place. ROS presents the illusion that certain versions of the model weights are stored and can be fetched on demand. Underneath, ROS does not physically store any copies of the weights; instead, it tracks the workers that hold these weights on GPUs for inference. Upon request, ROS directly uses them to serve reads. We build TensorHub, a production-quality system that instantiates the ROS idea with topology-aware transfer, model-parallel consistency, and fault tolerance. Evaluation shows that TensorHub saturates RDMA bandwidth and adapts to three distinct rollout workloads with minimal engineering effort. Specifically, TensorHub reduces total GPU stall time by up to 6.7x for standalone rollouts, accelerates weight updates for elastic rollouts by up to 4.8x, and cuts cross-datacenter rollout stall time by up to 19x. TensorHub has been deployed in ByteDance production to support cutting-edge RL training.
Sources
- Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedback
- Fast LLM Post-training via Decoupled and Fastest-of-N Speculation
- Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilities
- Safe RLHF: Safe Reinforcement Learning from Human Feedback
- AReaL: A Large-Scale Asynchronous Reinforcement Learning System for Language Reasoning
- RollPacker: Mitigating Long-Tail Rollouts for Fast, Synchronous RL Post-Training
- History Rhymes: Accelerating LLM Reinforcement Learning with RhymeRL
- OpenRLHF: An Easy-to-use, Scalable and High-performance RLHF Framework
- Taming the Long-Tail: Efficient Reasoning RL Training with Adaptive Drafter
- GPT-4o System Card
- ReaL: Efficient RLHF Training of Large Language Models with Parameter Reallocation
- Asynchronous RLHF: Faster and More Efficient Off-Policy RL for Language Models
- Proximal Policy Optimization Algorithms
- Rewarding Progress: Scaling Automated Process Verifiers for LLM Reasoning
- DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models
- SWE-RL: Advancing LLM Reasoning via Reinforcement Learning on Open Software Evolution
- LlamaRL: A Distributed Asynchronous Reinforcement Learning Framework for Efficient Large-scale LLM Training
- Learning to Reason under Off-Policy Guidance
- DAPO: An Open-Source LLM Reinforcement Learning System at Scale
- StreamRL: Scalable, Heterogeneous, and Elastic RL for LLMs with Disaggregated Stream Generation
Related papers
- iScheduler: Reinforcement Learning-Driven Continual Optimization for Large-Scale Resource Investment Problems
- SAMM: Sharded Automated Market Maker
- InferScale: GPU-Native KV Injection for Personalized LLM Serving
- Vigil: Accountable Liveness against Selective Silence
- Steelhead: Interleaving Partially Synchronous and Asynchronous Commit Rules on a Shared DAG
- Democratizing MoE inference on commodity GPUs with CoMoE