peterindia.net

Graph Neural Network (GNN) Frameworks

A verified directory of leading frameworks and libraries for graph analysis and Graph Neural Networks — covering PyTorch, TensorFlow, JAX, and Julia ecosystems.

10Frameworks listed
01

Plotly for Python

Plotly's Python graphing library makes interactive, publication-quality graphs — widely used for visualizing graph and network data.

Visit site
02

python-igraph

igraph is available on the Python Package Index with pre-compiled wheels for most Python distributions and platforms, for fast network analysis.

Visit site
03

NetworkX

A Python package for the creation, manipulation, and study of the structure, dynamics, and functions of complex networks.

Visit site
04

PyTorch Geometric (PyG)

A library built upon PyTorch to easily write and train Graph Neural Networks (GNNs) for a wide range of applications on structured data.

Visit site
05

Deep Graph Library (DGL)

Fast and memory-efficient message passing primitives for training Graph Neural Networks, framework-agnostic across PyTorch, TensorFlow, and MXNet.

Visit site
06

Graph Nets

DeepMind's library for building graph networks in TensorFlow and Sonnet, with demos for shortest-path, sorting, and physics prediction tasks.

Visit site
07

Spektral

A Python library for graph deep learning, based on the Keras API and TensorFlow 2, implementing popular GNN and pooling layers.

Visit site
08

GeometricFlux.jl

A framework for geometric deep learning in Julia, providing classic graph neural network layers and utility constructs built on Flux.

Visit site
09

Jraph

A lightweight library for working with graph neural networks in JAX, providing a graph data structure and a "zoo" of forkable GNN models.

Visit site
10

ptgnn

A PyTorch GNN library containing code for creating graph neural network models, with sample implementations for PPI, VarMisuse, and more.

Visit site