Ray Summit

Collective-on-Ray: High-performance Collective Communication for Distributed Machine Learning on Ray

Wednesday, June 23, 7:25PM UTC

Hao Zhang, Postdoc Scholar, UC Berkeley

View Slides >>>

Collective communication is the cornerstone of many distributed machine learning systems. While existing ML systems such as distributed TensorFlow have built-in collective communication infrastructure, these functionalities are often bundled with the framework, invisible to users, and cannot extrapolate to ML programs that are not written following the framework-specific languages.

In this talk, we introduce a set of Ray-native Python-based collective communication primitives for Ray clusters with distributed CPUs or GPUs. They can be used in Ray task or actor code to speed up distributed communications, such as those introduced in distributed ML training.

Built on top of these communication primitives, we bring in a Ray-native distributed ML training library, offering Python-based implementations and interfaces to a variety of data- and model-parallel training strategies, such as parameter server, pipeline parallelism, to enable training various DL or non-DL models, on a Ray cluster.


Hao Zhang

Hao Zhang

Postdoc Scholar, UC Berkeley

Hao Zhang is currently a postdoc scholar at RISE Lab, UC Berkeley. His research interest is majorly in scalable machine learning, deep learning, AutoML, and large-scale ML applications in computer vision and natural language processing. He co-designs models, algorithms and systems to scale out ML to larger data, problems and applications, to ease the prototyping of complex ML models and algorithms, and to automate the distribution of ML programs. Several of Hao's works including Poseidon, Cavs, and GeePS were parts of the Petuum project and the Ray project, and now being commercialized at several startup companies.