Graph Neural Networks Python Packages

torch-geometric

Graph Neural Network Library for PyTorch

jraph

A Graph Neural Network Library in Jax

openequivariance

OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.

dgl

Python package built to ease deep learning on graph, on top of existing DL frameworks.

recbole

A unified, comprehensive and efficient recommendation library

ogb

Benchmark datasets, data loaders, and evaluators for graph machine learning

pyg-nightly

Graph Neural Network Library for PyTorch

chgnet

Pretrained universal neural network potential for charge-informed atomistic modeling https://chgnet.lbl.gov

dgllife

Python package for graph neural networks in chemistry and biology

torch-cluster

PyTorch Extension Library of Optimized Graph Cluster Algorithms

dglgo

Python package built to ease deep learning on graph, on top of existing DL frameworks.

pyhgf

PyHGF: A neural network library for predictive coding

analysisg

A project which aims to abstract the analysis pipeline of a Particle Physics Analysis at ATLAS.

torch-spline-conv

Implementation of the Spline-Based Convolution Operator of SplineCNN in PyTorch

spektral

Graph Neural Networks with Keras and Tensorflow 2.

torch-geometric-temporal

PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)

ktrain

ktrain is a Python library that makes deep learning and AI more accessible and easier to apply

dress-graph

DRESS: A Continuous Framework for Structural Graph Refinement

graphein

Protein Graph Library

netgraph3d

Interactive 3D Network Visualization for Graphs

coverage-control

Environment for coverage control and learning using GNN

deeprobust

A pytorch adversarial library for attack and defense methods on images and graphs

pygod

A Python Library for Graph Outlier Detection (Anomaly Detection)

dgl-cu92

Python package built to ease deep learning on graph, on top of existing DL frameworks.