arXiv · 2605.12396
NCCLZ: Compression-Enabled GPU Collectives with Decoupled Quantization and Entropy Coding
Abstract
Collective communication is a major bottleneck for multi-node GPU workloads in scientific computing and distributed deep learning, especially when inter-node bandwidth is limited. Although NCCL provides optimized GPU-centric collectives, large messages can still dominate end-to-end performance. Existing compression-enabled collective libraries either rely on MPI-based stacks that cannot fully exploit NCCL, omit entropy coding, or tightly couple full compressors with communication primitives, limiting compression ratio, flexibility, and communication-computation overlap. This paper presents NCCLZ, a compression-enabled GPU collectives that decouples quantization and entropy coding and integrates them at different layers of the stack. NCCLZ places quantization at the interface, embeds entropy coding into NCCL primitives, uses a lightweight device-side selector to choose coding strategies, and overlaps compression with communication to reduce exposed overhead. Experiments on scientific datasets, training gradients, and synthetic workloads show up to 9.65x speedup over NCCL and up to 3.34x improvement over prior compression-assisted collective libraries.
Explore related subjects
Keep this discovery
Jiamin Wang, Zhijing Ye, Xiaodong Yu. 2026-05-12. NCCLZ: Compression-Enabled GPU Collectives with Decoupled Quantization and Entropy Coding. https://arxiv.org/abs/2605.12396
Cite the original work for its findings. Save a collection to share your selection of sources.