Recurrent Residual Module for Fast Inference in Videos. - Bowen Pan; Wuwei Lin; Xiaolin Fang 002; Chaoqin Huang; Bolei Zhou; Cewu Lu (CVPR, 2018)

Cross-Stream Selective Networks for Action Recognition. - Bowen Pan; Jiankai Sun; Wuwei Lin; Limin Wang 002; Weiyao Lin (CVPR Workshops, 2019)

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)

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)

FlashInfer: Efficient and Customizable Attention Engine for LLM Inference Serving. - Zihao Ye 001; Lequn Chen 001; Ruihang Lai; Wuwei Lin; Yineng Zhang; Stephanie Wang; Tianqi Chen 001; Baris Kasikci; Vinod Grover; Arvind Krishnamurthy; Luis Ceze (CoRR, 2025)