publications

2026

1. OSDI 26

   MPK: A Compiler and Runtime for Mega-Kernelizing Tensor Programs

   Xinhao Cheng^*, Zhihao Zhang^*, Yu Zhou^*, Jianan Ji^*, Jinchen Jiang, Zepeng Zhao, Ziruo Xiao, Zihao Ye, Yingyi Huang, Ruihang Lai, Hongyi Jin, Bohan Hou, Mengdi Wu, Yixin Dong, Anthony Yip, Zihao Ye, Songting Wang, Wenqin Yang, Xupeng Miao, Tianqi Chen, and Zhihao Jia

   In 20th USENIX Symposium on Operating Systems Design and Implementation (OSDI 26), Jul 2026

   arXiv Bib HTML Code

   @inproceedings{cheng2026mpk,
     title = {{MPK}: A Compiler and Runtime for {Mega-Kernelizing} Tensor Programs},
     author = {Cheng, Xinhao and Zhang, Zhihao and Zhou, Yu and Ji, Jianan and Jiang, Jinchen and Zhao, Zepeng and Xiao, Ziruo and Ye, Zihao and Huang, Yingyi and Lai, Ruihang and Jin, Hongyi and Hou, Bohan and Wu, Mengdi and Dong, Yixin and Yip, Anthony and Ye, Zihao and Wang, Songting and Yang, Wenqin and Miao, Xupeng and Chen, Tianqi and Jia, Zhihao},
     booktitle = {20th USENIX Symposium on Operating Systems Design and Implementation (OSDI 26)},
     year = {2026},
     month = jul,
     address = {Seattle, WA},
     pages = {1909--1926},
     isbn = {978-1-939133-55-7},
     publisher = {USENIX Association},
   }

2025

1. OSDI 25

   Mirage: A Multi-Level Superoptimizer for Tensor Programs

   Mengdi Wu, Xinhao Cheng, Shengyu Liu, Chunan Shi, Jianan Ji, Kit Ao, Praveen Velliengiri, Xupeng Miao, Oded Padon, and Zhihao Jia

   Jul 2025
2. ICDE 25

   DaVinci Sketch: A Versatile Sketch for Efficient and Comprehensive Set Measurements

   Yanshu Wang^*, Jianan Ji^*, Chao-Hsuan Liu, Hengyang Zhou, and Tong Yang

   In 2025 IEEE 41st International Conference on Data Engineering (ICDE), Jul 2025

2024

1. ICDE 24

   Bitmatcher: Bit-level counter adjustment for sketches

   Qilong Shi, Chengjun Jia, Wenjun Li, Zaoxing Liu, Tong Yang, Jianan Ji, Gaogang Xie, Weizhe Zhang, and Minlan Yu

   In 2024 IEEE 40th International Conference on Data Engineering (ICDE), Jul 2024

2023

1. SIGMOD 24

   Chainedfilter: Combining membership filters by chain rule

   Haoyu Li, Liuhui Wang, Qizhi Chen, Jianan Ji, Yuhan Wu, Yikai Zhao, Tong Yang, and Aditya Akella

   Proceedings of the ACM on Management of Data, Jul 2023