English

PyG 2.0: Scalable Learning on Real World Graphs

Machine Learning 2025-07-29 v2 Artificial Intelligence

Abstract

PyG (PyTorch Geometric) has evolved significantly since its initial release, establishing itself as a leading framework for Graph Neural Networks. In this paper, we present Pyg 2.0 (and its subsequent minor versions), a comprehensive update that introduces substantial improvements in scalability and real-world application capabilities. We detail the framework's enhanced architecture, including support for heterogeneous and temporal graphs, scalable feature/graph stores, and various optimizations, enabling researchers and practitioners to tackle large-scale graph learning problems efficiently. Over the recent years, PyG has been supporting graph learning in a large variety of application areas, which we will summarize, while providing a deep dive into the important areas of relational deep learning and large language modeling.

Keywords

Cite

@article{arxiv.2507.16991,
  title  = {PyG 2.0: Scalable Learning on Real World Graphs},
  author = {Matthias Fey and Jinu Sunil and Akihiro Nitta and Rishi Puri and Manan Shah and Blaž Stojanovič and Ramona Bendias and Alexandria Barghi and Vid Kocijan and Zecheng Zhang and Xinwei He and Jan Eric Lenssen and Jure Leskovec},
  journal= {arXiv preprint arXiv:2507.16991},
  year   = {2025}
}
R2 v1 2026-07-01T04:14:12.189Z