Yangtian Zhang

I am now a final year undergraduate in ACM Honors Class of Shanghai Jiao Tong University, advised by Prof. Weinan Zhang and Prof. Yong Yu. I'm also a research intern supervised by Prof. Jian Tang at Mila. Please check out my CV here.

Research Interests

My interest mainly lies in machine learning algorithm and its applications. I'm both interested in graph representation learning and reinforcement learning. I'm currently working on incorporating geometric information into graph neural network and its applications such as drug discovery.


Sept.2018-Present Shanghai Jiao Tong University
Pursuing Bachelor’s Degree of Computer Science, advised by Prof. Yong Yu.

Research Experience

Mar.2021-Present Mila

Aug.2020-Present Apex Data & Knowledge Management Lab, SJTU

Projects & Publications

TorchDrug: A powerful and flexible machine learning platform for drug discovery
Participated in building a powerful platform designed for drug discovery, covering techniques from graph machine learning (graph neural networks, geometric deep learning & knowledge graphs), deep generative models, deep sequence model to reinforcement learning.
See project here.
Automated Multi-Step Occupancy Measure Matching for Imitation Learning
Proposed an auto curriculum version of T-setp Occupancy Measure matching algorithm, AutoGAIL, that chooses the appropriate sequence length for sample efficiency.
See paper draft here.

Teaching Experience

Teaching Assistant of CS158: Data Structures and Algorithms(Honors Class) 2020
Teaching Assistant of MS125: Principle and Practice of Computer Algorithms(Honors Class) 2020
Teaching Assistant of CS420: Machine Learning 2021

Honor & Rewards

Zhiyuan Honorary Scholarship, (Top 5% in SJTU) 2018-2021
Excellent School-level Scholarship (Top 2% in SJTU) 2018-2021
Meritorious Winner in Mathematical Contest in Modeling (Top 8%) 2020
The First Prize in Chinese Physics Olympiad (Top 1%) 2017


I'm a true lover of steel-string acoustic guitar. I'm also good at Djembe and Cajon (a kind of drum). Currently I'm learning playing Pen Beat.
Besides machine learning, I'm also interested in physics and have won several awards in Chinese Physics Olympiad.