pyg-team / pytorch_geometric
Graph Neural Network Library for PyTorch
☆21,546Updated this week
Alternatives and similar repositories for pytorch_geometric:
Users that are interested in pytorch_geometric are comparing it to the libraries listed below
- Python package built to ease deep learning on graph, on top of existing DL frameworks.☆13,585Updated last month
- Must-read papers on graph neural networks (GNN)☆16,101Updated 11 months ago
- Graph Convolutional Networks in PyTorch☆5,213Updated 4 years ago
- Implementation of Graph Convolutional Networks in TensorFlow☆7,164Updated last year
- links to conference publications in graph-based deep learning☆4,842Updated 2 weeks ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,450Updated this week
- AI conference deadline countdowns☆5,697Updated 2 months ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆8,569Updated 3 weeks ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,885Updated 3 weeks ago
- Build Graph Nets in Tensorflow☆5,370Updated last year
- Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.☆28,592Updated this week
- A PyTorch implementation of the Transformer model in "Attention is All You Need".☆8,924Updated 7 months ago
- Paper Lists for Graph Neural Networks☆2,220Updated 11 months ago
- Repository for benchmarking graph neural networks☆2,534Updated last year
- Graph Attention Networks (https://arxiv.org/abs/1710.10903)☆3,256Updated 2 years ago
- PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)☆2,691Updated last month
- A curated list of awesome self-supervised methods☆6,183Updated 5 months ago
- The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.☆6,039Updated last month
- A scikit-learn compatible neural network library that wraps PyTorch☆5,903Updated last month
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆32,549Updated this week
- Model interpretability and understanding for PyTorch☆4,966Updated 2 weeks ago
- PyTorch tutorials.☆8,290Updated this week
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125☆14,368Updated this week
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,017Updated 9 months ago
- Datasets, Transforms and Models specific to Computer Vision☆16,356Updated this week
- PyTorch implementations of Generative Adversarial Networks.☆16,556Updated 5 months ago
- A Collection of Variational Autoencoders (VAE) in PyTorch.☆6,758Updated 5 months ago
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,553Updated this week
- Geometric Computer Vision Library for Spatial AI☆10,043Updated this week
- A collection of important graph embedding, classification and representation learning papers with implementations.☆4,764Updated last year