I am an Assistant Professor in the Machine Learning Department and Computer Science Department at Carnegie Mellon University . I help run the Catalyst Group. I am also the CTO of OctoML. I recently received my Ph.D. from Paul G. Allen School of Computer Science & Engineering, University of Washington.
At UW, I was advised by Carlos Guestrin. I also worked closely with Luis Ceze and Arvind Krishnamurthy. I was a member of SAMPL Lab and MODE Lab. I recieved my Master and Bachelor's degree from Prof. Yong Yu's ACM Honors Class at Shanghai Jiao Tong University (SJTU).
I am interested in the intersection of machine learning and systems. The real excitement of this area comes from what can enable when we bring advanced learning techniques and system together. On that end, I am also pushing the direction on deep learning, knowledge transfer and lifelong learning.
I am a believer of open source and open science. My research involves publishing algorithms in openly accessible mediums and building open-source machine learning systems that are widely adopted. Here are the ML systems that I created:
- Apache TVM, an Automated End-to-End Optimizing Compiler for Deep Learning.
- XGBoost, a scalable tree boosting system.
- Apache MXNet(co-creator)
- Google Ph.D. Fellowship
- Champion of KDDCup 2012
- Runner up of KDDCup 2011
Activity and Services
- Organizer of the LearningSys workshop NeurIPS 2015.
- Reviewer of ICML, IJCAI, NeurIPS, ICLR, JMLR, TIST
CMU, present Assistant Professor
OctoML, present CTO.
University of Washington, 2013 - 2019
Ph.D. student with Carlos Guestrin.
Google Brain. Summer 2015 Research Intern with Ian Goodfellow.
Graphlab Inc. Summer 2014 Intern, main author of boosted tree and neural net toolkit in Graphlab Create.
Apex Data and Knowledge Management Lab, Shanghai Jiao Tong University, China. 2010-2013 Master Student with Yong Yu
System Biology Lab, Shanghai Jiao Tong University. 2010 - 2013 Visiting Student, with Ping Ao
Digital Enterprise Research Institute, Galway, Ireland. Winter 2011 Visiting Scholar.
Microsoft Research Asia, Beijing, China. Summer 2009. Research Intern, with Jun Yan.