Ziheng Jiang

Ph.D. Student

University of Washington

Biography

Ziheng Jiang is a Ph.D. Student advised by Luis Ceze in the Paul G. Allen School of Computer Science & Engineering at the University of Washington. He also work closely with Tianqi Chen. His current research center around co-designing efficient algorithms and systems for machine learning. He received his Bachelor’s degree from Fudan University, where he was a member of Fudan NLP Lab, working with Xipeng Qiu, and Zheng Zhang.

Besides coding, he also enjoy snowboarding, photography, and badminton:)

Interests

Machine Learning
Large-Scale Computing System

Education

PhD in Computer Science, 2018 - Present
University of Washington
BSc in Computer Science, 2013 - 2018
Fudan University

Experience

Research Intern

Google Brain

Jun 2019 – Sep 2019 California

Graduate Research Assistant

University of Washington

Sep 2018 – Present Seattle

Research Intern

Amazon AWS Machine Learning Group

Feb 2017 – Oct 2017 California

Undergraduate Research Assistant

New York University, Shanghai

Apr 2016 – Jan 2017 Shanghai

Undergraduate Research Assistant

Fudan University

Sep 2015 – Jan 2017 Shanghai

Publications

[Preprint] Automated Backend-Aware Post-Training Quantization

Ziheng Jiang, Animesh Jain, Andrew Liu, Josh Fromm, Chengqian Ma, Tianqi Chen, Luis Ceze

PDF Preprint

[ICML 2021 (Long Oral)] Characterizing Structural Regularities of Labeled Data in Overparameterized Models

Ziheng Jiang, Chiyuan Zhang, Kunal Talwar, Michael C Mozer

PDF Preprint

[MICRO 2020 (Best Paper)] A Hardware–Software Blueprint for Flexible Deep Learning Specialization

Thierry Moreau, Tianqi Chen, Luis Vega, Jared Roesch, Eddie Yan, Lianmin Zheng, Josh Fromm, Ziheng Jiang, Luis Ceze, Carlos Guestrin, Arvind Krishnamurthy

PDF Webpage Preprint

[Preprint] Relay: A High-Level Compiler for Deep Learning

Jared Roesch, Steven Lyubomirsky, Marisa Kirisame, Logan Weber, Josh Pollock, Luis Vega, Ziheng Jiang, Tianqi Chen, Thierry Moreau, Zachary Tatlock

PDF Preprint

[NIPS 2018] Learning to Optimize Tensor Programs

Tianqi Chen, Lianmin Zheng, Eddie Yan, Ziheng Jiang, Thierry Moreau, Luis Ceze, Carlos Guestrin, Arvind Krishnamurthy

PDF Webpage Preprint

[OSDI 2018] TVM: An Automated End-to-End Optimizing Compiler for Deep Learning

Tianqi Chen, Thierry Moreau, Ziheng Jiang, Lianmin Zheng, Eddie Yan, Meghan Cowan, Leyuan Wang, Yuwei Hu, Luis Ceze, Carlos Guestrin, Arvind Krishnamurthy

PDF Webpage Preprint

[MLSys 2018] Efficient Deep Learning Inference on Edge Devices

Ziheng Jiang, Tianqi Chen, Mu Li

PDF Webpage