Ziheng Jiang

Ph.D. Student

University of Washington


Ziheng Jiang is a Ph.D. Student with Luis Ceze in the Paul G. Allen School of Computer Science & Engineering at the University of Washington. 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:)


  • Machine Learning
  • Large-Scale Computing System


  • PhD in Computer Science, 2018 - Present

    University of Washington

  • BSc in Computer Science, 2013 - 2018

    Fudan University

Recent Publications

Exploring the Memorization Generalization Continuum in Deep Learning

Human learners appreciate that some facts demand memorization whereas other facts support generalization. For example, English verbs …

A Hardware–Software Blueprint for Flexible Deep Learning Specialization

This article describes the Versatile Tensor Accelerator (VTA), a programmable DL architecture designed to be extensible in the face of …

Just in Time Dynamic Batching

Batching is an essential technique to improve computation efficiency in deep learning frameworks. While batch processing for models …

TVM: An Automated End-to-End Optimizing Compiler for Deep Learning

There is an increasing need to bring machine learning to a wide diversity of hardware devices. Current frameworks rely on …

Learning to Optimize Tensor Programs

We introduce a learning-based framework to optimize tensor programs for deep learning workloads. Efficient implementations of tensor …



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

Oct 2017 – Feb 2017 California

Undergraduate Research Assistant

New York University, Shanghai

Apr 2016 – Jan 2017 Shanghai

Undergraduate Research Assistant

Fudan University

Sep 2015 – Jan 2017 Shanghai