Rishi Puri

Joined

May 2025

Names

Rishi Puri (Preferred)

Emails

****@nvidia.com (Confirmed)

Personal Links

Career & Education History

Researcher

NVIDIA (nvidia.com)

2021 – 2025

Undergrad student

Electrical Engineering & Computer Science Department, University of California, Berkeley (eecs.berkeley.edu)

2016 – 2020

Advisors, Relations & Conflicts

Coauthor

2022 – 2025

Expertise

GNN+LLM

2024 – 2025

GNNs

2023 – 2025

Deep Learning

2019 – 2025

Publications

Relational Graph Transformer
Vijay Prakash Dwivedi, Sri Jaladi, Yangyi Shen, Federico Lopez, Charilaos I. Kanatsoulis, Rishi Puri, Matthias Fey, Jure Leskovec
- ICLR 2026 Poster
- Readers: Everyone
PyG 2.0: Scalable Learning on Real World Graphs
Matthias Fey, Jinu Sunil, Akihiro Nitta, Rishi Puri, Manan Shah, Blaž Stojanovič, Ramona Bendias, Alexandria Barghi, Vid Kocijan, Zecheng Zhang, Xinwei He, Jan Eric Lenssen, Jure Leskovec
- TGL @ KDD 2025
- Readers: Everyone
Query-Aware Graph Neural Networks for Enhanced Retrieval-Augmented Generation
Vibhor Agrawal, Fay Wang, Rishi Puri
- MLoG-GenAI@KDD Oral
- Readers: Everyone
Relational Graph Transformer
Vijay Prakash Dwivedi, Sri Jaladi, Yangyi Shen, Federico Lopez, Charilaos I. Kanatsoulis, Rishi Puri, Matthias Fey, Jure Leskovec
- TGL @ KDD 2025
- Readers: Everyone

Co-Authors