2D-HRA: Two-Dimensional Hierarchical Ring-Based All-Reduce Algorithm in Large-Scale Distributed Machine Learning. - Youhe Jiang; Huaxi Gu; Yunfeng Lu; Xiaoshan Yu (IEEE Access, 2020)

OSDP: Optimal Sharded Data Parallel for Distributed Deep Learning. - Youhe Jiang; Xupeng Miao; Xiaonan Nie; Bin Cui 001 (CoRR, 2022)

Galvatron: Efficient Transformer Training over Multiple GPUs Using Automatic Parallelism. - Xupeng Miao; Yujie Wang; Youhe Jiang; Chunan Shi; Xiaonan Nie; Hailin Zhang 004; Bin Cui 001 (Proc. VLDB Endow., 2022)

OSDP: Optimal Sharded Data Parallel for Distributed Deep Learning. - Youhe Jiang; Fangcheng Fu; Xupeng Miao; Xiaonan Nie; Bin Cui 001 (IJCAI, 2023)

Improving Automatic Parallel Training via Balanced Memory Workload Optimization. - Yujie Wang; Youhe Jiang; Xupeng Miao; Fangcheng Fu; Xiaonan Nie; Bin Cui 001 (CoRR, 2023)

HexGen: Generative Inference of Foundation Model over Heterogeneous Decentralized Environment. - Youhe Jiang; Ran Yan; Xiaozhe Yao; Beidi Chen; Binhang Yuan (CoRR, 2023)

HexGen: Generative Inference of Large Language Model over Heterogeneous Environment. - Youhe Jiang; Ran Yan; Xiaozhe Yao; Yang Zhou; Beidi Chen; Binhang Yuan (ICML, 2024)

GNNFingers: A Fingerprinting Framework for Verifying Ownerships of Graph Neural Networks. - Xiaoyu You; Youhe Jiang; Jianwei Xu; Mi Zhang 001; Min Yang 002 (WWW, 2024)

FlashFlex: Accommodating Large Language Model Training over Heterogeneous Environment. - Ran Yan; Youhe Jiang; Wangcheng Tao; Xiaonan Nie; Bin Cui 001; Binhang Yuan (CoRR, 2024)

Revisiting the Time Cost Model of AllReduce. - Dian Xiong; Li Chen 008; Youhe Jiang; Dan Li 001; Shuai Wang 028; Songtao Wang (CoRR, 2024)

Improving Automatic Parallel Training via Balanced Memory Workload Optimization. - Yujie Wang; Youhe Jiang; Xupeng Miao; Fangcheng Fu; Shenhan Zhu; Xiaonan Nie; Yaofeng Tu; Bin Cui 001 (IEEE Trans. Knowl. Data Eng., 2024)

HexGen-2: Disaggregated Generative Inference of LLMs in Heterogeneous Environment. - Youhe Jiang; Ran Yan; Binhang Yuan (ICLR, 2025)

Demystifying Cost-Efficiency in LLM Serving over Heterogeneous GPUs. - Youhe Jiang; Fangcheng Fu; Xiaozhe Yao; Guoliang He; Xupeng Miao; Ana Klimovic; Bin Cui 001; Binhang Yuan; Eiko Yoneki (CoRR, 2025)

ThunderServe: High-performance and Cost-efficient LLM Serving in Cloud Environments. - Youhe Jiang; Fangcheng Fu; Xiaozhe Yao; Taiyi Wang; Bin Cui 001; Ana Klimovic; Eiko Yoneki (CoRR, 2025)

HEXGEN-TEXT2SQL: Optimizing LLM Inference Request Scheduling for Agentic Text-to-SQL Workflow. - You Peng; Youhe Jiang; Chen Wang; Binhang Yuan (CoRR, 2025)

Thinking Short and Right Over Thinking Long: Serving LLM Reasoning Efficiently and Accurately. - Yuhang Wang; Youhe Jiang; Bin Cui 001; Fangcheng Fu (CoRR, 2025)

Cascadia: A Cascade Serving System for Large Language Models. - Youhe Jiang; Fangcheng Fu; Wanru Zhao; Stephan Rabanser; Nicholas D. Lane; Binhang Yuan (CoRR, 2025)