about
PyG 2.0: Scalable Learning on Real World Graphs (arxiv.org)
10 points by PaulHoule on Aug 16, 2025 | hide | past | pdf | 1 comment on HN

In plain words: PyG 2.0 is a major update to a toolkit for training graph neural networks, adding support for graphs with mixed node and edge types and for graphs that change over time. It keeps graph data on disk, so it can handle real-world graphs efficiently.

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.

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, et al.
arXiv:2507.16991 · cs.LG, cs.AI · submitted Jul 22, 2025 · updated Jul 27, 2025
abstract · pdf · html

add comment on HN