About me

I am currently a third year Ph.D. student in the Department of Electrical and Computer Engineering, Cornell University @ Computer System Lab, advised by Prof. Zhiru Zhang.

My research interests fall on building novel computer systems and supporting heterogeneous computing on novel applications, including computational genomics and machine learning. My recent focus is to leverage the best performance on GPUs.

I received my B.E. from Department of Electronic Engineering, Tsinghua University in 2020. I conducted my undergraduate research in NICS-EFC Lab, advised by Prof. Yu Wang. During the summer 2019, I was an research intern at UCLA VAST Lab, advised by Prof. Jason Cong.

Education

  • Cornell University
    • Ph.D. student in Electrical and Computer Engineering, starting from Sep. 2021.
  • Tsinghua University
    • B.E. in Electronic Engineering, Sep. 2016 to Jun. 2020.

Experience

  • ByteDance
    • Research Scientist Intern in Machine Learning System team in Seed Foundation, Aug 2024 to Dec 2024.
  • NVIDIA
    • Deep Learning Training Performance Intern in End-to-End Training Performance team (working on MLPerf-Training), May 2024 to Aug 2024.
  • Tsinghua University
  • Alibaba DAMO Academy
  • UCLA
    • Research Intern in VAST Lab, Jun. 2019 to Sep. 2019.

Publications

Rapid GPU-Based Pangenome Graph Layout
Jiajie Li, Jan-Niklas Schmelzle, Yixiao Du, Simon Heumos, Andrea Guarracino, Giulia Guidi, Pjotr Prins, Erik Garrison, Zhiru Zhang.
[SC’24]. Accepted and to be shown!

Pangenome graph layout by Path-Guided Stochastic Gradient Descent
Simon Heumos, Andrea Guarracino, Jan-Niklas M Schmelzle, Jiajie Li, Zhiru Zhang, Jörg Hagmann, Sven Nahnsen, Pjotr Prins, Erik Garrison.
[Bioinformatics]. Volume 40, Issue 7, July 2024.

NPS: A Framework for Accurate Program Sampling Using Graph Neural Network
Yuanwei Fang, Zihao Liu, Yanheng Lu, Jiawei Liu, Jiajie Li, Yi Jin, Jian Chen, Yenkuang Chen, Hongzhong Zheng, Yuan Xie.
arXiv:2304.08880, 2023.

Analysis and Optimization of GNN-Based Recommender Systems on Persistent Memory
Yuwei Hu, Jiajie Li, Zhongming Yu, Zhiru Zhang.
arXiv:2207.11918, 2022.

HeteroHalide: From Image Processing DSL to Efficient FPGA Acceleration
Jiajie Li, Yuze Chi, Jason Cong.
[FPGA’20]. 28th ACM/SIGDA International Symposium on Field-Programmable Gate Arrays, 2020.

Teaching

  • Teaching Assistant of ECE2300 Digital Logic and Computer Organization, Fall 2023.

Services