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)

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

Pseudo-triplet Guided Few-shot Composed Image Retrieval. - Bohan Hou; Haoqiang Lin; Haokun Wen; Meng Liu 006; Xuemeng Song (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)

FiRE: Enhancing MLLMs with Fine-Grained Context Learning for Complex Image Retrieval. - Bohan Hou; Haoqiang Lin; Xuemeng Song; Haokun Wen; Meng Liu 006; Yupeng Hu; Xiangyu Zhao 001 (SIGIR, 2025)

A Comprehensive Survey on Composed Image Retrieval. - Xuemeng Song; Haoqiang Lin; Haokun Wen; Bohan Hou; Mingzhu Xu; Liqiang Nie (CoRR, 2025)

Tilus: A Virtual Machine for Arbitrary Low-Precision GPGPU Computation in LLM Serving. - Yaoyao Ding; Bohan Hou; Xiao Zhang; Allan Lin; Tianqi Chen 001; Cody Yu Hao; Yida Wang 003; Gennady Pekhimenko (CoRR, 2025)

ImgEdit: A Unified Image Editing Dataset and Benchmark. - Yang Ye; Xianyi He; Zongjian Li; Bin Lin 014; Shenghai Yuan; Zhiyuan Yan; Bohan Hou; Li Yuan 007 (CoRR, 2025)