Chronus: A Novel Deadline-aware Scheduler for Deep Learning Training Jobs. - Wei Gao 064; Zhisheng Ye 002; Peng Sun 006; Yonggang Wen 001; Tianwei Zhang 004 (SoCC, 2021)

Privacy-Preserving Collaborative Learning With Automatic Transformation Search. - Wei Gao 064; Shangwei Guo; Tianwei Zhang 004; Han Qiu 001; Yonggang Wen 001; Yang Liu 003 (CVPR, 2021)

Titan: a scheduler for foundation model fine-tuning workloads. - Wei Gao 064; Peng Sun 006; Yonggang Wen 001; Tianwei Zhang 004 (SoCC, 2022)

Deep Learning Workload Scheduling in GPU Datacenters: Taxonomy, Challenges and Vision. - Wei Gao 064; Qinghao Hu 004; Zhisheng Ye 002; Peng Sun 006; Xiaolin Wang 001; Yingwei Luo; Tianwei Zhang 004; Yonggang Wen 001 (CoRR, 2022)

Astraea: A Fair Deep Learning Scheduler for Multi-Tenant GPU Clusters. - Zhisheng Ye 002; Peng Sun 006; Wei Gao 064; Tianwei Zhang 004; Xiaolin Wang 001; Shengen Yan; Yingwei Luo (IEEE Trans. Parallel Distributed Syst., 2022)

Automatic Transformation Search Against Deep Leakage From Gradients. - Wei Gao 064; Xu Zhang; Shangwei Guo; Tianwei Zhang 004; Tao Xiang 001; Han Qiu 001; Yonggang Wen 001; Yang Liu 003 (IEEE Trans. Pattern Anal. Mach. Intell., 2023)

AutoSched: An Adaptive Self-configured Framework for Scheduling Deep Learning Training Workloads. - Wei Gao 064; Xu Zhang; Shan Huang; Shangwei Guo; Peng Sun 006; Yonggang Wen 001; Tianwei Zhang 004 (ICS, 2024)

Ymir: A Scheduler for Foundation Model Fine-tuning Workloads in Datacenters. - Wei Gao 064; Weiming Zhuang; Minghao Li 005; Peng Sun 006; Yonggang Wen 001; Tianwei Zhang 004 (ICS, 2024)

A Layer-Elastic Scheduling System for DLT in GPU Clusters. - Wei Gao 064 (Zenodo, 2024)

Deep Learning Workload Scheduling in GPU Datacenters: A Survey. - Zhisheng Ye 002; Wei Gao 064; Qinghao Hu 004; Peng Sun 006; Xiaolin Wang 001; Yingwei Luo; Tianwei Zhang 004; Yonggang Wen 001 (ACM Comput. Surv., 2024)

UniSched: A Unified Scheduler for Deep Learning Training Jobs With Different User Demands. - Wei Gao 064; Zhisheng Ye 002; Peng Sun 006; Tianwei Zhang 004; Yonggang Wen 001 (IEEE Trans. Computers, 2024)

SPPO:Efficient Long-sequence LLM Training via Adaptive Sequence Pipeline Parallel Offloading. - Qiaoling Chen; Shenggui Li; Wei Gao 064; Peng Sun 006; Yonggang Wen 001; Tianwei Zhang 004 (CoRR, 2025)

Rethinking Key-Value Cache Compression Techniques for Large Language Model Serving. - Wei Gao 064; Xinyu Zhou; Peng Sun 006; Tianwei Zhang 004; Yonggang Wen 001 (CoRR, 2025)

IceFrog: A Layer-Elastic Scheduling System for Deep Learning Training in GPU Clusters. - Wei Gao 064; Zhuoyuan Ouyang; Peng Sun 006; Tianwei Zhang 004; Yonggang Wen 001 (IEEE Trans. Parallel Distributed Syst., 2025)

AdaPtis: Reducing Pipeline Bubbles with Adaptive Pipeline Parallelism on Heterogeneous Models. - Jihu Guo; Tenghui Ma; Wei Gao 064; Peng Sun 006; Jiaxing Li; Xun Chen; Yuyang Jin 001; Dahua Lin (CoRR, 2025)