2022 SparseTIR: Composable Abstractions for Sparse Compilation in Deep Learning Zihao Ye, Ruihang Lai, Junru Shao, Tianqi Chen, Luis Ceze In submission. Under review. arXiv TensorIR: An Abstraction for Automatic Tensorized Program Optimization Siyuan Feng, Bohan Hou, Hongyi Jin, Wuwei Lin, Junru Shao, Ruihang Lai, Zihao Ye, Lianmin Zheng, Cody Hao Yu, Yong Yu, Tianqi Chen In submission. Under review. arXiv 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 Accepted by NeurIPS 2022. arXiv