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.

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

recbole

A unified, comprehensive and efficient recommendation library

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.

analysisg

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

pyhgf

PyHGF: A neural network library for predictive coding

torch-spline-conv

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

spektral

Graph Neural Networks with Keras and Tensorflow 2.

ktrain

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

xnns

Neural network interatomic potentials for chemistry and materials science

graphein

Protein Graph Library

gnnpcsaftpropertypackage

GNNPCSAFT CAPE-OPEN Property Package

dgl-cu92

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

graphscope-client

🔨 🍇 💻 🚀 GraphScope: A One-Stop Large-Scale Graph Computing System from Alibaba | 一站式图计算系统

city2graph

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

gnnepcsaft

GNNPCSAFT Project

torch-geometric-temporal

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