Publications

Ziheng Jiang, Chiyuan Zhang, Kunal Talwar, Michael C Mozer (2020). Characterizing Structural Regularities of Labeled Data in Overparameterized Models.

PDF Preprint

Thierry Moreau, Tianqi Chen, Luis Vega, Jared Roesch, Eddie Yan, Lianmin Zheng, Josh Fromm, Ziheng Jiang, Luis Ceze, Carlos Guestrin, Arvind Krishnamurthy (2019). A hardware–software blueprint for flexible deep learning specialization. IEEE Micro.

PDF Webpage Preprint

Jared Roesch, Steven Lyubomirsky, Marisa Kirisame, Logan Weber, Josh Pollock, Luis Vega, Ziheng Jiang, Tianqi Chen, Thierry Moreau, Zachary Tatlock (2019). Relay: A High-Level Compiler for Deep Learning.

PDF Preprint

Tianqi Chen, Lianmin Zheng, Eddie Yan, Ziheng Jiang, Thierry Moreau, Luis Ceze, Carlos Guestrin, Arvind Krishnamurthy (2018). Learning to Optimize Tensor Programs. NIPS'18.

PDF Webpage Preprint

Tianqi Chen, Thierry Moreau, Ziheng Jiang, Lianmin Zheng, Eddie Yan, Meghan Cowan, Leyuan Wang, Yuwei Hu, Luis Ceze, Carlos Guestrin, Arvind Krishnamurthy (2018). TVM: An Automated End-to-End Optimizing Compiler for Deep Learning. OSDI'18.

PDF Webpage Preprint

Ziheng Jiang, Tianqi Chen, Mu Li (2018). Efficient Deep Learning Inference on Edge Devices. SysML'18.

PDF Webpage