Tensor Program Optimization with Probabilistic Programs. - Junru Shao; Xiyou Zhou; Siyuan Feng; Bohan Hou; Ruihang Lai; Hongyi Jin; Wuwei Lin; Masahiro Masuda; Cody Hao Yu; Tianqi Chen 001 (NeurIPS, 2022)

TensorIR: An Abstraction for Automatic Tensorized Program Optimization. - Siyuan Feng; Bohan Hou; Hongyi Jin; Wuwei Lin; Junru Shao; Ruihang Lai; Zihao Ye 001; Lianmin Zheng; Cody Hao Yu; Yong Yu 001; Tianqi Chen 001 (ASPLOS (2), 2023)

Towards Efficient Generative Large Language Model Serving: A Survey from Algorithms to Systems. - Xupeng Miao; Gabriele Oliaro; Zhihao Zhang 001; Xinhao Cheng; Hongyi Jin; Tianqi Chen 001; Zhihao Jia (CoRR, 2023)

A System for Microserving of LLMs. - Hongyi Jin; Ruihang Lai; Charlie F. Ruan; Yingcheng Wang; Todd C. Mowry; Xupeng Miao; Zhihao Jia; Tianqi Chen 001 (CoRR, 2024)

WebLLM: A High-Performance In-Browser LLM Inference Engine. - Charlie F. Ruan; Yucheng Qin; Xun Zhou; Ruihang Lai; Hongyi Jin; Yixin Dong; Bohan Hou; Meng-Shiun Yu; Yiyan Zhai; Sudeep Agarwal; Hangrui Cao; Siyuan Feng; et al. (CoRR, 2024)

Relax: Composable Abstractions for End-to-End Dynamic Machine Learning. - Ruihang Lai; Junru Shao; Siyuan Feng; Steven Lyubomirsky; Bohan Hou; Wuwei Lin; Zihao Ye 001; Hongyi Jin; Yuchen Jin; Jiawei Liu 004; Lesheng Jin; Yaxing Cai; et al. (ASPLOS (2), 2025)