Alternatives and similar repositories for pytorch_geometric

Users that are interested in pytorch_geometric are comparing it to the libraries listed below

• dmlc / dgl
  Python package built to ease deep learning on graph, on top of existing DL frameworks.
  ☆13,874Updated 3 months ago
• tkipf / pygcn
  Graph Convolutional Networks in PyTorch
  ☆5,304Updated 4 years ago
• tkipf / gcn
  Implementation of Graph Convolutional Networks in TensorFlow
  ☆7,257Updated 2 years ago
• thunlp / GNNPapers
  Must-read papers on graph neural networks (GNN)
  ☆16,398Updated last year
• naganandy / graph-based-deep-learning-literature
  links to conference publications in graph-based deep learning
  ☆4,916Updated this week
• google-deepmind / graph_nets
  Build Graph Nets in Tensorflow
  ☆5,384Updated 2 years ago
• lanpa / tensorboardX
  tensorboard for pytorch (and chainer, mxnet, numpy, ...)
  ☆7,935Updated 2 weeks ago
• graphdeeplearning / benchmarking-gnns
  Repository for benchmarking graph neural networks (JMLR 2023)
  ☆2,586Updated last year
• PetarV- / GAT
  Graph Attention Networks (https://arxiv.org/abs/1710.10903)
  ☆3,352Updated 3 years ago
• pytorch / captum
  Model interpretability and understanding for PyTorch
  ☆5,209Updated last week
• huggingface / pytorch-image-models
  The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…
  ☆34,105Updated this week
• pytorch / examples
  A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
  ☆23,043Updated this week
• NVIDIA / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆8,650Updated last week
• arogozhnikov / einops
  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
  ☆8,903Updated 3 weeks ago
• yunjey / pytorch-tutorial
  PyTorch Tutorial for Deep Learning Researchers
  ☆31,258Updated last year
• williamleif / GraphSAGE
  Representation learning on large graphs using stochastic graph convolutions.
  ☆3,518Updated 9 months ago
• facebookresearch / detr
  End-to-End Object Detection with Transformers
  ☆14,331Updated last year
• microsoft / nni
  An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model c…
  ☆14,186Updated 10 months ago
• google-research / vision_transformer
  ☆11,336Updated 2 months ago
• skorch-dev / skorch
  A scikit-learn compatible neural network library that wraps PyTorch
  ☆6,021Updated 3 weeks ago
• AntixK / PyTorch-VAE
  A Collection of Variational Autoencoders (VAE) in PyTorch.
  ☆7,135Updated last month
• TrustAGI-Lab / Awesome-Graph-Neural-Networks
  Paper Lists for Graph Neural Networks
  ☆2,257Updated last year
• Lightning-AI / pytorch-lightning
  Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
  ☆29,439Updated last week
• microsoft / Swin-Transformer
  This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
  ☆14,755Updated 9 months ago
• optuna / optuna
  A hyperparameter optimization framework
  ☆11,910Updated last week
• jadore801120 / attention-is-all-you-need-pytorch
  A PyTorch implementation of the Transformer model in "Attention is All You Need".
  ☆9,177Updated last year
• pytorch / vision
  Datasets, Transforms and Models specific to Computer Vision
  ☆16,824Updated this week
• facebookresearch / PyTorch-BigGraph
  Generate embeddings from large-scale graph-structured data.
  ☆3,412Updated last year
• sksq96 / pytorch-summary
  Model summary in PyTorch similar to `model.summary()` in Keras
  ☆4,040Updated last year
• snap-stanford / ogb
  Benchmark datasets, data loaders, and evaluators for graph machine learning
  ☆2,001Updated last week