HET: Scaling out Huge Embedding Model Training via Cache-enabled Distributed Framework. - Xupeng Miao; Hailin Zhang 004; Yining Shi 001; Xiaonan Nie; Zhi Yang 001; Yangyu Tao; Bin Cui 001 (Proc. VLDB Endow., 2021)

HET-GMP: A Graph-based System Approach to Scaling Large Embedding Model Training. - Xupeng Miao; Yining Shi 001; Hailin Zhang 004; Xin Zhang; Xiaonan Nie; Zhi Yang 001; Bin Cui 001 (SIGMOD Conference, 2022)

Welder: Scheduling Deep Learning Memory Access via Tile-graph. - Yining Shi 001; Zhi Yang 001; Jilong Xue; Lingxiao Ma; Yuqing Xia; Ziming Miao; Yuxiao Guo 001; Fan Yang 024; Lidong Zhou (OSDI, 2023)

Cocktailer: Analyzing and Optimizing Dynamic Control Flow in Deep Learning. - Chen Zhang 001; Lingxiao Ma; Jilong Xue; Yining Shi 001; Ziming Miao; Fan Yang 024; Jidong Zhai; Zhi Yang 001; Mao Yang 004 (OSDI, 2023)

SDPipe: A Semi-Decentralized Framework for Heterogeneity-aware Pipeline-parallel Training. - Xupeng Miao; Yining Shi 001; Zhi Yang 001; Bin Cui 001; Zhihao Jia (Proc. VLDB Endow., 2023)

Ladder: Enabling Efficient Low-Precision Deep Learning Computing through Hardware-aware Tensor Transformation. - Lei Wang 222; Lingxiao Ma; Shijie Cao; Quanlu Zhang; Jilong Xue; Yining Shi 001; Ningxin Zheng; Ziming Miao; Fan Yang 024; Ting Cao; Yuqing Yang 001; Mao Yang 004 (OSDI, 2024)

PipeThreader: Software-Defined Pipelining for Efficient DNN Execution. - Yu Cheng; Lei Wang 222; Yining Shi 001; Yuqing Xia; Lingxiao Ma; Jilong Xue; Yang Wang 053; Zhiwen Mo; Feiyang Chen; Fan Yang 024; Mao Yang 004; Zhi Yang 001 (OSDI, 2025)

TileLang: A Composable Tiled Programming Model for AI Systems. - Lei Wang 222; Yu Cheng; Yining Shi 001; Zhengju Tang; Zhiwen Mo; Wenhao Xie; Lingxiao Ma; Yuqing Xia; Jilong Xue; Fan Yang 024; Zhi Yang 001 (CoRR, 2025)

Efficient and scalable huge embedding model training via distributed cache management. - Xupeng Miao; Hailin Zhang 004; Yining Shi 001; Xiaonan Nie; Zhi Yang 001; Yangyu Tao; Jie Jiang; Bin Cui 001 (VLDB J., 2025)