PyG 2.0: Scalable Learning on Real World Graphs

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Fey, Matthias, Sunil, Jinu, Nitta, Akihiro, Puri, Rishi, Shah, Manan, Stojanovič, Blaž, Bendias, Ramona, Barghi, Alexandria, Kocijan, Vid, Zhang, Zecheng, He, Xinwei, Lenssen, Jan Eric, Leskovec, Jure
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913961832611840
author Fey, Matthias
Sunil, Jinu
Nitta, Akihiro
Puri, Rishi
Shah, Manan
Stojanovič, Blaž
Bendias, Ramona
Barghi, Alexandria
Kocijan, Vid
Zhang, Zecheng
He, Xinwei
Lenssen, Jan Eric
Leskovec, Jure
author_facet Fey, Matthias
Sunil, Jinu
Nitta, Akihiro
Puri, Rishi
Shah, Manan
Stojanovič, Blaž
Bendias, Ramona
Barghi, Alexandria
Kocijan, Vid
Zhang, Zecheng
He, Xinwei
Lenssen, Jan Eric
Leskovec, Jure
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2507_16991
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle PyG 2.0: Scalable Learning on Real World Graphs
Fey, Matthias
Sunil, Jinu
Nitta, Akihiro
Puri, Rishi
Shah, Manan
Stojanovič, Blaž
Bendias, Ramona
Barghi, Alexandria
Kocijan, Vid
Zhang, Zecheng
He, Xinwei
Lenssen, Jan Eric
Leskovec, Jure
Machine Learning
Artificial Intelligence
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.
title PyG 2.0: Scalable Learning on Real World Graphs
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2507.16991