Titan AI LogoTitan AI

pytorch_geometric

22,893
3,885
Python

Project Description

Graph Neural Network Library for PyTorch

pytorch_geometric: Graph Neural Network Library for PyTorch

Project Title

pytorch_geometric โ€” Graph Neural Network Library for PyTorch

Overview

PyTorch Geometric (PyG) is a library built upon PyTorch for easily writing and training Graph Neural Networks (GNNs) for a wide range of applications related to structured data. It offers a unified API, comprehensive GNN models, and great flexibility for research and real-world applications.

Key Features

  • Unified and easy-to-use API for training GNN models
  • Comprehensive and well-maintained GNN models
  • Great flexibility for extending existing models or creating new architectures
  • Support for large-scale real-world GNN models

Use Cases

  • Machine learning researchers using GNNs for structured data applications
  • Developers looking for a PyTorch-based library for graph neural networks
  • Researchers and practitioners working with large-scale graph data

Advantages

  • Utilizes a tensor-centric API, keeping design principles close to vanilla PyTorch
  • Easy-to-use message passing API for creating custom GNN architectures
  • Supports learning on diverse types of graphs, including scalable, dynamic, and heterogeneous GNNs

Limitations / Considerations

  • License information is unknown, which may affect usage in certain projects
  • As with any specialized library, there may be a learning curve for new users

Similar / Related Projects

  • DGL (Deep Graph Library): A Python package built to ease deep learning on graph-structured data. It differs in its focus on distributed training and heterogeneous graph support.
  • Spektral: A Python library for graph deep learning that provides a Keras-like API. It is designed for users familiar with Keras and TensorFlow.

Basic Information


๐Ÿ“Š Project Information

  • Project Name: pytorch_geometric
  • GitHub URL: https://github.com/pyg-team/pytorch_geometric
  • Programming Language: Python
  • โญ Stars: 22,778
  • ๐Ÿด Forks: 3,871
  • ๐Ÿ“… Created: 2017-10-06
  • ๐Ÿ”„ Last Updated: 2025-08-20

๐Ÿท๏ธ Project Topics

Topics: [, ", d, e, e, p, -, l, e, a, r, n, i, n, g, ", ,, , ", g, e, o, m, e, t, r, i, c, -, d, e, e, p, -, l, e, a, r, n, i, n, g, ", ,, , ", g, r, a, p, h, -, c, o, n, v, o, l, u, t, i, o, n, a, l, -, n, e, t, w, o, r, k, s, ", ,, , ", g, r, a, p, h, -, n, e, u, r, a, l, -, n, e, t, w, o, r, k, s, ", ,, , ", p, y, t, o, r, c, h, ", ]


๐Ÿ“š Documentation


This article is automatically generated by AI based on GitHub project information and README content analysis

Titan AI Explorehttps://www.titanaiexplore.com/projects/pytorch_geometric-106024057en-USTechnology

Project Information

Created on 10/6/2017
Updated on 9/16/2025