PAI-FCNN: FPGA Based Inference System for Complex CNN Models. - Lixue Xia; Lansong Diao; Zhao Jiang; Hao Liang 003; Kai Chen 008; Li Ding; Shunli Dou; Zibin Su; Meng Sun; Jiansong Zhang 001; Wei Lin 016 (ASAP, 2019)

PAI-FCNN: FPGA Based CNN Inference System. - Lansong Diao; Zhao Jiang; Hao Liang 003; Chang'an Ye; Kai Chen 008; Li Ding; Shunli Dou; Meng Sun; Lixue Xia; Jiansong Zhang 001; Wei Lin 016 (FPGA, 2019)

Optimizing distributed training deployment in heterogeneous GPU clusters. - Xiaodong Yi 001; Shiwei Zhang 002; Ziyue Luo; Guoping Long; Lansong Diao; Chuan Wu 001; Zhen Zheng; Jun Yang 052; Wei Lin 016 (CoNEXT, 2020)

Auto-MAP: A DQN Framework for Exploring Distributed Execution Plans for DNN Workloads. - Siyu Wang; Yi Rong; Shiqing Fan; Zhen Zheng; Lansong Diao; Guoping Long; Jun Yang 052; Xiaoyong Liu; Wei Lin 016 (CoRR, 2020)

FusionStitching: Boosting Memory Intensive Computations for Deep Learning Workloads. - Zhen Zheng; Pengzhan Zhao; Guoping Long; Feiwen Zhu; Kai Zhu; Wenyi Zhao; Lansong Diao; Jun Yang 052; Wei Lin 016 (CoRR, 2020)

DISC: A Dynamic Shape Compiler for Machine Learning Workloads. - Kai Zhu; Wenyi Zhao; Zhen Zheng; Tianyou Guo; Pengzhan Zhao; Junjie Bai; Jun Yang 052; Xiaoyong Liu; Lansong Diao; Wei Lin 016 (EuroMLSys@EuroSys, 2021)

DAPPLE: a pipelined data parallel approach for training large models. - Shiqing Fan; Yi Rong; Chen Meng; Zongyan Cao; Siyu Wang; Zhen Zheng; Chuan Wu 001; Guoping Long; Jun Yang 052; Lixue Xia; Lansong Diao; Xiaoyong Liu; et al. (PPoPP, 2021)

Accelerating large-scale distributed neural network training with SPMD parallelism. - Shiwei Zhang 002; Lansong Diao; Chuan Wu 001; Siyu Wang; Wei Lin 016 (SoCC, 2022)

Optimizing DNN Compilation for Distributed Training With Joint OP and Tensor Fusion. - Xiaodong Yi 001; Shiwei Zhang 002; Lansong Diao; Chuan Wu 001; Zhen Zheng; Shiqing Fan; Siyu Wang; Jun Yang 052; Wei Lin 016 (IEEE Trans. Parallel Distributed Syst., 2022)

Auto-Parallelizing Large Models with Rhino: A Systematic Approach on Production AI Platform. - Shiwei Zhang 002; Lansong Diao; Siyu Wang; Zongyan Cao; Yiliang Gu; Chang Si; Ziji Shi; Zhen Zheng; Chuan Wu 001; Wei Lin 016 (CoRR, 2023)

Ada-Grouper: Accelerating Pipeline Parallelism in Preempted Network by Adaptive Group-Scheduling for Micro-Batches. - Siyu Wang; Zongyan Cao; Chang Si; Lansong Diao; Jiamang Wang; Wei Lin 016 (CoRR, 2023)

Expediting Distributed DNN Training With Device Topology-Aware Graph Deployment. - Shiwei Zhang 002; Xiaodong Yi 001; Lansong Diao; Chuan Wu 001; Siyu Wang; Wei Lin 016 (IEEE Trans. Parallel Distributed Syst., 2023)

HAP: SPMD DNN Training on Heterogeneous GPU Clusters with Automated Program Synthesis. - Shiwei Zhang 002; Lansong Diao; Chuan Wu 001; Zongyan Cao; Siyu Wang; Wei Lin 016 (Zenodo, 2023)

HAP: SPMD DNN Training on Heterogeneous GPU Clusters with Automated Program Synthesis. - Shiwei Zhang 002; Lansong Diao; Chuan Wu 001; Zongyan Cao; Siyu Wang; Wei Lin 016 (Zenodo, 2023)

FaPES: Enabling Efficient Elastic Scaling for Serverless Machine Learning Platforms. - Xiaoyang Zhao 005; Siran Yang; Jiamang Wang; Lansong Diao; Lin Qu; Chuan Wu 001 (SoCC, 2024)

HAP: SPMD DNN Training on Heterogeneous GPU Clusters with Automated Program Synthesis. - Shiwei Zhang 002; Lansong Diao; Chuan Wu 001; Zongyan Cao; Siyu Wang; Wei Lin 016 (EuroSys, 2024)

Concerto: Automatic Communication Optimization and Scheduling for Large-Scale Deep Learning. - Shenggan Cheng; Shengjie Lin; Lansong Diao; Hao Wu 077; Siyu Wang; Chang Si; Ziming Liu; Xuanlei Zhao; Jiangsu Du; Wei Lin 016; Yang You 001 (ASPLOS (1), 2025)

SRDiffusion: Accelerate Video Diffusion Inference via Sketching-Rendering Cooperation. - Shenggan Cheng; Yuanxin Wei; Lansong Diao; Yong Liu 020; Bujiao Chen; Lianghua Huang; Yu Liu 023; Wenyuan Yu; Jiangsu Du; Wei Lin 016; Yang You 001 (CoRR, 2025)