Spindle: Informed Memory Access Monitoring. - Haojie Wang 004; Jidong Zhai; Xiongchao Tang; Bowen Yu 003; Xiaosong Ma; Wenguang Chen (USENIX ATC, 2018)

Spread-n-share: improving application performance and cluster throughput with resource-aware job placement. - Xiongchao Tang; Haojie Wang 004; Xiaosong Ma; Nosayba El-Sayed; Jidong Zhai; Wenguang Chen; Ashraf Aboulnaga (SC, 2019)

Identifying scalability bottlenecks for large-scale parallel programs with graph analysis. - Yuyang Jin 001; Haojie Wang 004; Xiongchao Tang; Torsten Hoefler; Xu Liu 001; Jidong Zhai (PPoPP, 2020)

ScalAna: automating scaling loss detection with graph analysis. - Yuyang Jin 001; Haojie Wang 004; Teng Yu; Xiongchao Tang; Torsten Hoefler; Xu Liu 001; Jidong Zhai (SC, 2020)

Sparker: Efficient Reduction for More Scalable Machine Learning with Spark. - Bowen Yu 003; Huanqi Cao; Tianyi Shan; Haojie Wang 004; Xiongchao Tang; Wenguang Chen (ICPP, 2021)

HyQuas: hybrid partitioner based quantum circuit simulation system on GPU. - Chen Zhang 001; Zeyu Song; Haojie Wang 004; Kaiyuan Rong; Jidong Zhai (ICS, 2021)

PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections. - Haojie Wang 004; Jidong Zhai; Mingyu Gao 001; Zixuan Ma; Shizhi Tang; Liyan Zheng 001; Yuanzhi Li; Kaiyuan Rong; Yuanyong Chen; Zhihao Jia (OSDI, 2021)

LotusSQL: SQL engine for high-performance big data systems. - Xiaohan Li; Bowen Yu 003; Guanyu Feng; Haojie Wang 004; Wenguang Chen (Big Data Min. Anal., 2021)

Chukonu: A Fully-Featured Big Data Processing System by Efficiently Integrating a Native Compute Engine into Spark. - Bowen Yu 003; Guanyu Feng; Huanqi Cao; Xiaohan Li; Zhenbo Sun; Haojie Wang 004; Xiaowei Zhu 001; Weimin Zheng; Wenguang Chen (Proc. VLDB Endow., 2021)

Efficiently emulating high-bitwidth computation with low-bitwidth hardware. - Zixuan Ma; Haojie Wang 004; Guanyu Feng; Chen Zhang 001; Lei Xie; Jiaao He; Shengqi Chen 001; Jidong Zhai (ICS, 2022)

FreeTensor: a free-form DSL with holistic optimizations for irregular tensor programs. - Shizhi Tang; Jidong Zhai; Haojie Wang 004; Lin Jiang; Liyan Zheng 001; Zhenhao Yuan; Chen Zhang 001 (PLDI, 2022)

FasterMoE: modeling and optimizing training of large-scale dynamic pre-trained models. - Jiaao He; Jidong Zhai; Tiago Antunes; Haojie Wang 004; Fuwen Luo; Shangfeng Shi; Qin Li (PPoPP, 2022)

PerFlow: a domain specific framework for automatic performance analysis of parallel applications. - Yuyang Jin 001; Haojie Wang 004; Runxin Zhong; Chen Zhang 001; Jidong Zhai (PPoPP, 2022)

BaGuaLu: targeting brain scale pretrained models with over 37 million cores. - Zixuan Ma; Jiaao He; Jiezhong Qiu; Huanqi Cao; Yuanwei Wang; Zhenbo Sun; Liyan Zheng 001; Haojie Wang 004; Shizhi Tang; Tianyu Zheng; Junyang Lin; Guanyu Feng; et al. (PPoPP, 2022)

Vapro: performance variance detection and diagnosis for production-run parallel applications. - Liyan Zheng 001; Jidong Zhai; Xiongchao Tang; Haojie Wang 004; Teng Yu; Yuyang Jin 001; Shuaiwen Leon Song; Wenguang Chen (PPoPP, 2022)

UniQ: A Unified Programming Model for Efficient Quantum Circuit Simulation. - Chen Zhang 001; Haojie Wang 004; Zixuan Ma; Lei Xie; Zeyu Song; Jidong Zhai (SC, 2022)

OLLIE: Derivation-based Tensor Program Optimizer. - Liyan Zheng 001; Haojie Wang 004; Jidong Zhai; Muyan Hu; Zixuan Ma; Tuowei Wang; Shizhi Tang; Lei Xie; Kezhao Huang; Zhihao Jia (CoRR, 2022)

Detecting Performance Variance for Parallel Applications Without Source Code. - Jidong Zhai; Liyan Zheng 001; Feng Zhang 007; Xiongchao Tang; Haojie Wang 004; Teng Yu; Yuyang Jin 001; Shuaiwen Leon Song; Wenguang Chen (IEEE Trans. Parallel Distributed Syst., 2022)

EINNET: Optimizing Tensor Programs with Derivation-Based Transformations. - Liyan Zheng 001; Haojie Wang 004; Jidong Zhai; Muyan Hu; Zixuan Ma; Tuowei Wang; Shuhong Huang; Xupeng Miao; Shizhi Tang; Kezhao Huang; Zhihao Jia (OSDI, 2023)

GraphSet: High Performance Graph Mining through Equivalent Set Transformations. - Tianhui Shi; Jidong Zhai; Haojie Wang 004; Qiqian Chen; Mingshu Zhai; Zixu Hao; Haoyu Yang; Wenguang Chen (SC, 2023)

PowerFusion: A Tensor Compiler with Explicit Data Movement Description and Instruction-level Graph IR. - Zixuan Ma; Haojie Wang 004; Jingze Xing; Liyan Zheng 001; Chen Zhang 001; Huanqi Cao; Kezhao Huang; Shizhi Tang; Penghan Wang; Jidong Zhai (CoRR, 2023)

Unified Programming Models for Heterogeneous High-Performance Computers. - Zixuan Ma; Yuyang Jin 001; Shizhi Tang; Haojie Wang 004; Wei-Cheng Xue; Jidong Zhai; Wei-Min Zheng (J. Comput. Sci. Technol., 2023)

Optimizing DNNs With Partially Equivalent Transformations and Automated Corrections. - Haojie Wang 004; Jidong Zhai; Mingyu Gao 001; Feng Zhang 007; Tuowei Wang; Zixuan Ma; Shizhi Tang; Liyan Zheng 001; Wen Wang; Kaiyuan Rong; Yuanyong Chen; Zhihao Jia (IEEE Trans. Computers, 2023)

Optimal Kernel Orchestration for Tensor Programs with Korch. - Muyan Hu; Ashwin Venkatram; Shreyashri Biswas; Balamurugan Marimuthu; Bohan Hou; Gabriele Oliaro; Haojie Wang 004; Liyan Zheng 001; Xupeng Miao; Jidong Zhai; Zhihao Jia (ASPLOS (3), 2024)

AdaPipe: Optimizing Pipeline Parallelism with Adaptive Recomputation and Partitioning. - Zhenbo Sun; Huanqi Cao; Yuanwei Wang; Guanyu Feng; Shengqi Chen 001; Haojie Wang 004; Wenguang Chen (ASPLOS (3), 2024)

WiseGraph: Optimizing GNN with Joint Workload Partition of Graph and Operations. - Kezhao Huang; Jidong Zhai; Liyan Zheng 001; Haojie Wang 004; Yuyang Jin 001; Qihao Zhang; Runqing Zhang; Zhen Zheng; Youngmin Yi; Xipeng Shen (EuroSys, 2024)

MAGPY: Compiling Eager Mode DNN Programs by Monitoring Execution States. - Chen Zhang 001; Rongchao Dong; Haojie Wang 004; Runxin Zhong; Jike Chen; Jidong Zhai (USENIX ATC, 2024)

Graph-Centric Performance Analysis for Large-Scale Parallel Applications. - Yuyang Jin 001; Haojie Wang 004; Runxin Zhong; Chen Zhang 001; Xia Liao; Feng Zhang 007; Jidong Zhai (IEEE Trans. Parallel Distributed Syst., 2024)

IntelliGen: Instruction-Level Auto-tuning for Tensor Program with Monotonic Memory Optimization. - Zixuan Ma; Haojie Wang 004; Jingze Xing; Shuhong Huang; Liyan Zheng 001; Chen Zhang 001; Huanqi Cao; Kezhao Huang; Mingshu Zhai; Shizhi Tang; Penghan Wang; Jidong Zhai (CGO, 2025)

Leveraging Graph Analysis to Pinpoint Root Causes of Scalability Issues for Parallel Applications. - Yuyang Jin 001; Haojie Wang 004; Xiongchao Tang; Zhenhua Guo 003; Yaqian Zhao; Torsten Hoefler; Tao Liu 029; Xu Liu 001; Jidong Zhai (IEEE Trans. Parallel Distributed Syst., 2025)

FastCache: Optimizing Multimodal LLM Serving through Lightweight KV-Cache Compression Framework. - Jianian Zhu; Hang Wu; Haojie Wang 004; Yinghui Li; Biao Hou; Ruixuan Li 001; Jidong Zhai (CoRR, 2025)

SpecRouter: Adaptive Routing for Multi-Level Speculative Decoding in Large Language Models. - Hang Wu; Jianian Zhu; Yinghui Li; Haojie Wang 004; Biao Hou; Jidong Zhai (CoRR, 2025)

Taming Dynamic Diffusion LLM Inference through Virtual Static Execution. - Jianian Zhu; Hang Wu; Yinghui Li; Haojie Wang 004; Ruixuan Li 001; Jidong Zhai (ICS, 2026)

ChituDiffusion: A Data-Characteristic-Aware Serving System for Diffusion Models. - Chengzhang Wu; Liyan Zheng 001; Haojie Wang 004; Kezhao Huang; Zixuan Ma; Dong Dong 001; Jidong Zhai (PPoPP, 2026)

UniOrch: A Unified Mixed Framework for High-Efficiency LLM Training on Heterogeneous AI Chips. - Jingyi Wang; Jia Wang 038; Jun Gao; Yan Cao; Yang Zhai; Haojie Wang 004; Wanxin Song; Dezhi Kong; Liu Liu; Wei Li 008; Zhaofeng He 001 (IEEE Trans. Parallel Distributed Syst., 2026)