Publications

*: indicating equal contribution.

Optimal Kernel Orchestration for Tensor Programs with Korch

Muyan Hu, Ashwin Venkatram∗, Shreyashri Biswas∗, Balamurugan Marimuthu∗, Bohan Hou, Gabriele Oliaro, Haojie Wang, Liyan Zheng, Xupeng Miao, Jidong Zhai, Zhihao Jia

Proceedings of the 29th ACM International Conference on Architectural Support for Programming Languages and Operating Systems (ASPLOS’24)

EfficientViT: Lightweight Multi-Scale Attention for High-Resolution Dense Prediction

Han Cai, Junyan Li, Muyan Hu, Chuang Gan, Song Han

Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV), 2023

Media coverage: MIT Homepage

EinNet: Optimizing Tensor Programs with Derivation-Based Transformations

Liyan Zheng, Haojie Wang, Jidong Zhai, Muyan Hu, Zixuan Ma, Tuowei Wang, Shuhong Huang, Xupeng Miao, Shizhi Tang, Kezhao Huang, Zhihao Jia

17th USENIX Symposium on Operating Systems Design and Implementation (OSDI’23)