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

21k stars
252 watching
4k forks
Language: Python
last commit: 9 days ago
Linked from 6 awesome lists

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

Backlinks from these awesome lists:

Related projects:

Repository Description Stars
diego999/pygat An implementation of the Graph Attention Network model using PyTorch. 2,914
benedekrozemberczki/pytorch_geometric_temporal A PyTorch extension for building temporal graph neural networks with support for recurrent and attention-based models 2,677
tkipf/pygcn A PyTorch implementation of a graph neural network architecture for semi-supervised classification 5,198
pytorch/pytorch A Python library providing tensors and dynamic neural networks with strong GPU acceleration 84,371
yunjey/pytorch-tutorial A comprehensive tutorial project that provides code examples for learning PyTorch by implementing various deep learning models and demonstrating their usage. 30,286
sherylhyx/pytorch_geometric_signed_directed A PyTorch Geometric extension library for working with signed and directed graphs 128
chingyaoc/ggnn.pytorch An implementation of a neural network architecture for processing graph-structured data and making predictions on nodes. 465
dsgiitr/d2l-pytorch An open-source implementation of the popular book 'Dive Into Deep Learning' in PyTorch 4,244
dmlc/dgl A Python package for building deep learning models on graph data structures 13,548
bupt-gamma/openhgnn An open-source toolkit for training and applying heterogeneous graph neural networks using PyTorch and the Deep Graph Library. 867
facebookresearch/pytorch3d A deep learning library for 3D data processing and computer vision research using PyTorch 8,824
pytorch/examples A collection of curated examples showcasing various PyTorch applications in computer vision, natural language processing, and reinforcement learning. 22,428
graphdeeplearning/benchmarking-gnns A collection of datasets and tools for evaluating graph neural networks 2,523
szagoruyko/pytorchviz A tool to visualize PyTorch execution graphs and traces 3,228
google-deepmind/graph_nets An open-source library for building graph networks in TensorFlow and Sonnet 5,360