🤖 AI Summary
To address critical bottlenecks—including low communication throughput, high latency, and poor scalability—in training and inference of large language models (LLMs) on GPU clusters exceeding 100,000 devices, this paper introduces NCCLX, the first high-performance collective communication framework specifically designed for ultra-large-scale LLMs. NCCLX features: (1) a hierarchical, topology-aware communication scheduler that adapts to heterogeneous network architectures; (2) a dynamic bandwidth optimization mechanism to maximize link utilization; and (3) hardware-coordinated fault-tolerant transmission, ensuring reliability at million-node scale. Evaluated on the Llama4 model across 100,000 GPUs, NCCLX achieves a 2.3× improvement in end-to-end communication efficiency and reduces latency by 41%, outperforming all existing industry solutions. These advances establish NCCLX as a scalable, robust communication infrastructure foundation for both ultra-large-model training and low-latency inference.
📝 Abstract
The increasing scale of large language models (LLMs) necessitates highly efficient collective communication frameworks, particularly as training workloads extend to hundreds of thousands of GPUs. Traditional communication methods face significant throughput and latency limitations at this scale, hindering both the development and deployment of state-of-the-art models. This paper presents the NCCLX collective communication framework, developed at Meta, engineered to optimize performance across the full LLM lifecycle, from the synchronous demands of large-scale training to the low-latency requirements of inference. The framework is designed to support complex workloads on clusters exceeding 100,000 GPUs, ensuring reliable, high-throughput, and low-latency data exchange. Empirical evaluation on the Llama4 model demonstrates substantial improvements in communication efficiency. This research contributes a robust solution for enabling the next generation of LLMs to operate at unprecedented scales.