RAT - Resilient Allreduce Tree for Distributed Machine Learning. - Xinchen Wan; Hong Zhang 025; Hao Wang 116; Shuihai Hu; Junxue Zhang 001; Kai Chen 005 (APNet, 2020)

Domain-specific Communication Optimization for Distributed DNN Training. - Hao Wang 116; Jingrong Chen 004; Xinchen Wan; Han Tian; Jiacheng Xia; Gaoxiong Zeng; Weiyan Wang; Kai Chen 005; Wei Bai 001; Junchen Jiang (CoRR, 2020)

TACC: A Full-stack Cloud Computing Infrastructure for Machine Learning Tasks. - Kaiqiang Xu; Xinchen Wan; Hao Wang 116; Zhenghang Ren; Xudong Liao; Decang Sun; Chaoliang Zeng; Kai Chen 005 (CoRR, 2021)

Automatic Configuration for Optimal Communication Scheduling in DNN Training. - Yiqing Ma; Hao Wang 116; Yiming Zhang 003; Kai Chen 005 (CoRR, 2021)

Herald: An Embedding Scheduler for Distributed Embedding Model Training. - Chaoliang Zeng; Xiaodian Cheng; Han Tian; Hao Wang 116; Kai Chen 005 (APNet, 2022)

AutoByte: Automatic Configuration for Optimal Communication Scheduling in DNN Training. - Yiqing Ma; Hao Wang 116; Yiming Zhang 003; Kai Chen 005 (INFOCOM, 2022)

AutoPipe: Automatic Configuration of Pipeline Parallelism in Shared GPU Cluster. - Jinbin Hu; Ying Liu; Hao Wang 116; Jin Wang 001 (ICPP, 2024)

Towards Domain-Specific Network Transport for Distributed DNN Training. - Hao Wang 116; Han Tian; Jingrong Chen 004; Xinchen Wan; Jiacheng Xia; Gaoxiong Zeng; Wei Bai 001; Junchen Jiang; Yong Wang 046; Kai Chen 005 (NSDI, 2024)

Accelerating Neural Recommendation Training with Embedding Scheduling. - Chaoliang Zeng; Xudong Liao; Xiaodian Cheng; Han Tian; Xinchen Wan; Hao Wang 116; Kai Chen 005 (NSDI, 2024)

Design and Operation of Shared Machine Learning Clusters on Campus. - Kaiqiang Xu; Decang Sun; Hao Wang 116; Zhenghang Ren; Xinchen Wan; Xudong Liao; Zilong Wang 007; Junxue Zhang 001; Kai Chen 005 (ASPLOS (1), 2025)

Enabling In-Network Acceleration Over the Cloud. - Hao Wang 116; Decang Sun; Jinbin Hu; Kai Chen 005 (INFOCOM, 2025)

Towards Optimal Rack-scale μs-level CPU Scheduling through In-Network Workload Shaping. - Xudong Liao; Han Tian; Xinchen Wan; Chaoliang Zeng; Hao Wang 116; Junxue Zhang 001; Mengyu Ma; Guyue Liu; Kai Chen 005 (USENIX ATC, 2025)