Pytorch Geometric Python Packages

graphein

Protein Graph Library

mps-pointops

Point cloud ops (FPS, kNN, ball query) for PyTorch on Apple Silicon (MPS)

city2graph

Transform geospatial relations into graphs for Graph Neural Networks and spatial network analysis

gnnepcsaft

GNNPCSAFT Project

torch-geometric-signed-directed

PyTorch Geometric Signed Directed is a signed/directed graph neural network extension library for PyTorch Geometric. The paper is accepted by LoG 2023.

koopman-graph

Topology-aware Koopman autoencoders for forecasting and analyzing networked dynamics

mumin

Seamlessly build the MuMiN dataset.

heterosplit

Leakage-safe entity-disjoint splits and audits for heterogeneous link-prediction datasets (PyTorch Geometric-compatible).

graphchem

Molecular graph neural networks for predicting chemical properties, with a focus on fuels

fairmofsyncondition

A robust Python module for predicting the synthesis conditions of MOFs. It determines potential crystal structures and thermodynamic stability from a set of reagents or recommends optimal synthesis conditions based on given crystal structures.

falkordb-pyg

PyTorch Geometric Remote Backend for FalkorDB

syntherela

A package for benchmarking synthetic relational data generation methods

geome

Geometric Learning for Genome Data

graphssl

Modular PyTorch + PyG library for self-supervised learning on graphs: DGI, GraphCL, BGRL, AFGRL, GraphDINO, VICReg and Barlow Twins behind one config-driven API, with plain training loops (no Lightning, no Hydra) and reproducible benchmarks.

neurofhir

Geometric Deep Learning for Clinical Informatics: Converting FHIR to Temporal-Hierarchical Graph Tensors

slientruss3d

slientruss3d : Python for stable truss analysis and optimization tool

interpretable

interpretable:🧰toolkit for interpretable learning on graphs

greatx

A graph reliability toolbox based on PyTorch and PyTorch Geometric (PyG).

gigl

Gigantic Graph Learning (GiGL) Framework: Large-scale training and inference for Graph Neural Networks

graphwar

A graph reliability toolbox based on PyTorch and PyTorch Geometric (PyG).

fastgraphml

Given an input graph (ArangoDB or PyG) it generates graph embeddings using Low-Code framework built on top of PyG.

mooon

Graph Data Augmentation Library for PyTorch Geometric

graph-ood

GOOD: A Graph Out-of-Distribution Benchmark [NeurIPS 2022 Datasets and Benchmarks]