pytorch_geometric

Graph neural network library

A PyTorch-based library for training and applying Graph Neural Networks to structured data

Graph Neural Network Library for PyTorch

GitHub

22k stars
254 watching
4k forks
Language: Python
last commit: almost 2 years ago
Linked from 6 awesome lists

deep-learninggeometric-deep-learninggraph-convolutional-networksgraph-neural-networkspytorch

Backlinks from these awesome lists:

Related projects:

RepositoryDescriptionStars
diego999/pygatAn implementation of the Graph Attention Network model using PyTorch.2,935
benedekrozemberczki/pytorch_geometric_temporalA PyTorch extension for building temporal graph neural networks with support for recurrent and attention-based models2,694
tkipf/pygcnA PyTorch implementation of a graph neural network architecture for semi-supervised classification5,214
pytorch/pytorchA Python library providing tensors and dynamic neural networks with strong GPU acceleration84,978
yunjey/pytorch-tutorialA comprehensive tutorial project that provides code examples for learning PyTorch by implementing various deep learning models and demonstrating their usage.30,401
sherylhyx/pytorch_geometric_signed_directedA PyTorch Geometric extension library for working with signed and directed graphs131
chingyaoc/ggnn.pytorchAn implementation of a neural network architecture for processing graph-structured data and making predictions on nodes.466
dsgiitr/d2l-pytorchAn open-source implementation of the popular book 'Dive Into Deep Learning' in PyTorch4,256
dmlc/dglA Python package for building deep learning models on graph data structures13,601
bupt-gamma/openhgnnAn open-source toolkit for training and applying heterogeneous graph neural networks using PyTorch and the Deep Graph Library.879
facebookresearch/pytorch3dA deep learning library for 3D data processing and computer vision research using PyTorch8,889
pytorch/examplesA collection of curated examples showcasing various PyTorch applications in computer vision, natural language processing, and reinforcement learning.22,530
graphdeeplearning/benchmarking-gnnsA collection of datasets and tools for evaluating graph neural networks2,537
szagoruyko/pytorchvizA tool to visualize PyTorch execution graphs and traces3,251
google-deepmind/graph_netsBuilds graph networks in TensorFlow using a library provided by DeepMind's AI research organization.5,370