April 2022 - Present
AI Agent, Graph Neural Network (GNN), Explainability (XAI), Tabular Deep Learning, and AutoML
June 2021 - March 2022
Speech to Text, Automatic Speech Recognition (ASR), ML Infrastructure, and Language Models
2018 - 2020 Summers
Three summer internships working on Frontend, Database and Machine Learning
May 2024 - Present
Relational Deep Learning Benchmark: Open benchmark for machine learning over relational databases
April 2022 - Present
Graph Neural Network Library for PyTorch: A library built upon PyTorch to easily write and train Graph Neural Networks (GNNs) for a wide range of applications related to structured data
June 2023 - Present
August 2023 - Present
Tabular Deep Learning Library for PyTorch: A modular deep learning framework for building neural network models on heterogeneous tabular data
January 2020 - April 2022
Python library assists deep learning on graphs with flexible graph manipulation, standard pipeline, heterogeneous graphs and simple API.
Bowen Song, Zecheng ZhangZecheng Zhang, Zhaoxu Luo, Jason Hu, Wei Yuan, Jing Jia, Zhengxu Tang, Guanyang Wang and Liyue Shen. CCS: Controllable and Constrained Sampling with Diffusion Models via Initial Noise Perturbation CCS: Controllable and Constrained Sampling with Diffusion Models via Initial Noise Perturbation . (Arxiv 2025).
GenAI
Yiwen Yuan, Zecheng ZhangZecheng Zhang, Xinwei He, Akihiro Nitta, Weihua Hu, Dong Wang, Manan Shah, Shenyang Huang, Blaž Stojanovič, Alan Krumholz, Jan Eric Lenssen, Jure Leskovec and Matthias Fey. ContextGNN: Beyond Two-Tower Recommendation Systems ContextGNN: Beyond Two-Tower Recommendation Systems. In the International Conference on Learning Representations 2025 (ICLR 2025).
GNN RecSys
Tianqi Xu, Linyao Chen, Dai-Jie Wu, Yanjun Chen, Zecheng ZhangZecheng Zhang, Xiang Yao, Zhiqiang Xie, Yongchao Chen, Shilong Liu, Bochen Qian, Philip Torr, Bernard Ghanem and Guohao Li. CRAB: Cross-environment Agent Benchmark for Multimodal Language Model Agents CRAB: Cross-environment Agent Benchmark for Multimodal Language Model Agents. In the Open-World Agents Workshop @ NeurIPS 2024 (NeurIPS 2024 OWA Workshop).
Software Agent GenAI
Joshua Robinson*, Rishabh Ranjan*, Weihua Hu*, Kexin Huang*, Jiaqi Han, Alejandro Dobles, Matthias Fey, Jan E. Lenssen, Yiwen Yuan, Zecheng ZhangZecheng Zhang, Xinwei He and Jure Leskovec. RelBench: A Benchmark for Deep Learning on Relational Databases RelBench: A Benchmark for Deep Learning on Relational Databases. In the Annual Conference on Neural Information Processing Systems 2024 (NeurIPS 2024).
GNN RecSys Software
Weihua Hu, Yiwen Yuan, Zecheng ZhangZecheng Zhang, Akihiro Nitta, Kaidi Cao, Vid Kocijan, Jure Leskovec and Matthias Fey. PyTorch Frame: A Modular Framework for Multi-Modal Tabular Learning PyTorch Frame: A Modular Framework for Multi-Modal Tabular Learning. In the 3rd Table Representation Learning Workshop @ NeurIPS 2024 (NeurIPS 2024 TRL Workshop). Best Paper Award
GNN Tabular Software NLP
Bowen Song, Soo Min Kwon, Zecheng ZhangZecheng Zhang, Xinyu Hu, Qing Qu and Liyue Shen. Solving Inverse Problems with Latent Diffusion Models via Hard Data Consistency Solving Inverse Problems with Latent Diffusion Models via Hard Data Consistency. In the International Conference on Learning Representations 2024 (ICLR 2024).
GenAI
Yuxin Xiao, Zecheng ZhangZecheng Zhang, Yuning Mao, Carl Yang and Jiawei Han. SAIS: Supervising and Augmenting Intermediate Steps for Document-Level Relation ExtractionSAIS: Supervising and Augmenting Intermediate Steps for Document-Level Relation Extraction. In the 2022 Annual Conference of the North American Chapter of the Association for Computational Linguistics (NAACL 2022).
GNN NLP
Huajie Shao, Dachun Sun, Jiahao Wu, Zecheng ZhangZecheng Zhang, Aston Zhang, Shuochao Yao, Shengzhong Liu, Tianshi Wang, Chao Zhang and Tarek Abdelzaher. paper2repo: GitHub Repository Recommendation for Academic Papers paper2repo: GitHub Repository Recommendation for Academic Papers. In the Web Conference 2020 (WWW 2020).
GNN RecSys
Navjot Singh, Zecheng ZhangZecheng Zhang, Xiaoxiao Wu, Naijing Zhang, Siyuan Zhang and Edgar Solomonik. Enabling Distributed-Memory Tensor Completion in Python using New Sparse Tensor KernelsEnabling Distributed-Memory Tensor Completion in Python using New Sparse Tensor Kernels. In the Journal of Parallel and Distributed Computing (JPDC).
Software
Yuxin Xiao*, Zecheng ZhangZecheng Zhang*, Carl Yang and Chengxiang Zhai. Non-local Attention Learning on Large Heterogeneous Information Networks Non-local Attention Learning on Large Heterogeneous Information Networks. In the 2019 IEEE International Conference on Big Data (BigData 2019).
GNN
Email Me At
zecheng [at] cs [dot] stanford [dot] edu