CoT: Cooperative Training for Generative Modeling of Discrete Data. - Sidi Lu; Lantao Yu; Siyuan Feng 007; Yaoming Zhu; Weinan Zhang 001 (ICML, 2019)

CityFlow: A Multi-Agent Reinforcement Learning Environment for Large Scale City Traffic Scenario. - Huichu Zhang; Siyuan Feng 007; Chang Liu 021; Yaoyao Ding; Yichen Zhu 002; Zihan Zhou 002; Weinan Zhang 001; Yong Yu 001; Haiming Jin; Zhenhui Li (WWW, 2019)

Tensor Program Optimization with Probabilistic Programs. - Junru Shao; Xiyou Zhou; Siyuan Feng 007; 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 007; 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)

Effectively Scheduling Computational Graphs of Deep Neural Networks toward Their Domain-Specific Accelerators. - Jie Zhao 002; Siyuan Feng 007; Xiaoqiang Dan; Fei Liu; Chengke Wang; Sheng Yuan; Wenyuan Lv; Qikai Xie (OSDI, 2023)

Emerging Platforms Meet Emerging LLMs: A Year-Long Journey of Top-Down Development. - Siyuan Feng 007; Jiawei Liu 004; Ruihang Lai; Charlie F. Ruan; Yong Yu 001; Lingming Zhang 001; Tianqi Chen 001 (CoRR, 2024)

Vortex: Efficient Sample-Free Dynamic Tensor Program Optimization via Hardware-aware Strategy Space Hierarchization. - Yangjie Zhou 001; Honglin Zhu; Qian Qiu; Weihao Cui; Zihan Liu 002; Cong Guo 003; Siyuan Feng 007; Jintao Meng 001; Haidong Lan; Jingwen Leng; Wenxi Zhu; Minwen Deng (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 007; et al. (CoRR, 2024)

Relax: Composable Abstractions for End-to-End Dynamic Machine Learning. - Ruihang Lai; Junru Shao; Siyuan Feng 007; 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)

Magneto: Accelerating Parallel Structures in DNNs via Co-Optimization of Operators. - Zhanyuan Di; Leping Wang; Ziyi Ren; En Shao; Jie Zhao 002; Siyuan Feng 007; Dingwen Tao; Guangming Tan; Ninghui Sun (PPoPP, 2025)

A Sample-Free Compilation Framework for Efficient Dynamic Tensor Computation. - Yangjie Zhou 001; Honglin Zhu; Qian Qiu; Weihao Cui; Zihan Liu 002; Peng Chen 035; Mohamed Wahib; Cong Guo 003; Siyuan Feng 007; Jintao Meng 001; Haidong Lan; Jingwen Leng; et al. (SC, 2025)

Productively Deploying Emerging Models on Emerging Platforms: A Top-Down Approach for Testing and Debugging. - Siyuan Feng 007; Jiawei Liu 004; Ruihang Lai; Charlie Ruan; Yong Yu 001; Lingming Zhang 001; Tianqi Chen 001 (Proc. ACM Softw. Eng., 2025)

Accelerating Parallel Structures in DNNs via Parallel Fusion and Operator Co-Optimization. - Zhanyuan Di; Leping Wang; Zhaojia Ma; En Shao; Jie Zhao 002; Ziyi Ren; Siyuan Feng 007; Dingwen Tao; Guangming Tan; Ninghui Sun (ACM Trans. Archit. Code Optim., 2025)

FalconGEMM: Surpassing Hardware Peaks with Lower-Complexity Matrix Multiplication. - Honglin Zhu; Jiaping Cao; Jiang Shao; Siyuan Feng 007; Qian Qiu; Peng Chen 035; Xu Zhang 080; Yixian Zhou; Man Lung Yiu; Guang Ji; Minwen Deng; Wenxi Zhu; et al. (CoRR, 2026)