Elastic deep learning in multi-tenant GPU cluster. - Yidi Wu; Kaihao Ma; Xiao Yan 002; Zhi Liu; James Cheng (CoRR, 2019)

DGCL: an efficient communication library for distributed GNN training. - Zhenkun Cai; Xiao Yan 002; Yidi Wu; Kaihao Ma; James Cheng; Fan Yu (EuroSys, 2021)

Seastar: vertex-centric programming for graph neural networks. - Yidi Wu; Kaihao Ma; Zhenkun Cai; Tatiana Jin; Boyang Li 016; Chengguang Zheng; James Cheng; Fan Yu (EuroSys, 2021)

TensorOpt: Exploring the Tradeoffs in Distributed DNN Training With Auto-Parallelism. - Zhenkun Cai; Xiao Yan 002; Kaihao Ma; Yidi Wu; Yuzhen Huang; James Cheng; Teng Su; Fan Yu (IEEE Trans. Parallel Distributed Syst., 2022)

Elastic Deep Learning in Multi-Tenant GPU Clusters. - Yidi Wu; Kaihao Ma; Xiao Yan 002; Zhi Liu; Zhenkun Cai; Yuzhen Huang; James Cheng; Han Yuan; Fan Yu (IEEE Trans. Parallel Distributed Syst., 2022)

FEC: Efficient Deep Recommendation Model Training with Flexible Embedding Communication. - Kaihao Ma; Xiao Yan 002; Zhenkun Cai; Yuzhen Huang; Yidi Wu; James Cheng (Proc. ACM Manag. Data, 2023)

PPS: Fair and efficient black-box scheduling for multi-tenant GPU clusters. - Kaihao Ma; Zhenkun Cai; Xiao Yan 002; Yang Zhang; Zhi Liu; Yihui Feng; Chao Li; Wei Lin; James Cheng (Parallel Comput., 2024)

Adaptive Parallel Training for Graph Neural Networks. - Kaihao Ma; Renjie Liu 001; Xiao Yan 002; Zhenkun Cai; Xiang Song 003; Minjie Wang; Yichao Li; James Cheng (PPoPP, 2025)

KBest: Efficient Vector Search on Kunpeng CPU. - Kaihao Ma; Meiling Wang; Senkevich Oleg; Zijian Li; Daihao Xue; Dmitriy Malyshev; Yangming Lv; Shihai Xiao; Xiao Yan; Radionov Alexander; Weidi Zeng; Yuanzhan Gao; et al. (CoRR, 2025)