pyg-team / pytorch_geometric

Related projects ⓘ

Alternatives and complementary repositories for pytorch_geometric

• dmlc / dgl
  Python package built to ease deep learning on graph, on top of existing DL frameworks.
  ☆13,552Updated last month
• tkipf / gcn
  Implementation of Graph Convolutional Networks in TensorFlow
  ☆7,146Updated last year
• thunlp / GNNPapers
  Must-read papers on graph neural networks (GNN)
  ☆16,042Updated 11 months ago
• tkipf / pygcn
  Graph Convolutional Networks in PyTorch
  ☆5,200Updated 4 years ago
• google-deepmind / graph_nets
  Build Graph Nets in Tensorflow
  ☆5,362Updated last year
• naganandy / graph-based-deep-learning-literature
  links to conference publications in graph-based deep learning
  ☆4,815Updated this week
• Lightning-AI / pytorch-lightning
  Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
  ☆28,423Updated this week
• NVIDIA / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆8,415Updated 2 weeks ago
• graphdeeplearning / benchmarking-gnns
  Repository for benchmarking graph neural networks
  ☆2,525Updated last year
• pytorch / examples
  A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
  ☆22,439Updated last week
• lanpa / tensorboardX
  tensorboard for pytorch (and chainer, mxnet, numpy, ...)
  ☆7,871Updated this week
• facebookresearch / detr
  End-to-End Object Detection with Transformers
  ☆13,626Updated 8 months ago
• pytorch / captum
  Model interpretability and understanding for PyTorch
  ☆4,935Updated this week
• pytorch / ignite
  High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
  ☆4,528Updated last week
• PetarV- / GAT
  Graph Attention Networks (https://arxiv.org/abs/1710.10903)
  ☆3,237Updated 2 years ago
• TrustAGI-Lab / Awesome-Graph-Neural-Networks
  Paper Lists for Graph Neural Networks
  ☆2,214Updated 10 months ago
• optuna / optuna
  A hyperparameter optimization framework
  ☆10,939Updated this week
• arogozhnikov / einops
  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
  ☆8,524Updated this week
• fossasia / visdom
  A flexible tool for creating, organizing, and sharing visualizations of live, rich data. Supports Torch and Numpy.
  ☆10,043Updated 5 months ago
• facebookresearch / PyTorch-BigGraph
  Generate embeddings from large-scale graph-structured data.
  ☆3,387Updated 8 months ago
• bharathgs / Awesome-pytorch-list
  A comprehensive list of pytorch related content on github,such as different models,implementations,helper libraries,tutorials etc.
  ☆15,478Updated 9 months ago
• victoresque / pytorch-template
  PyTorch deep learning projects made easy.
  ☆4,762Updated 5 months ago
• Diego999 / pyGAT
  Pytorch implementation of the Graph Attention Network model by Veličković et. al (2017, https://arxiv.org/abs/1710.10903)
  ☆2,917Updated last year
• benedekrozemberczki / pytorch_geometric_temporal
  PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)
  ☆2,674Updated last month
• facebookresearch / pytorch3d
  PyTorch3D is FAIR's library of reusable components for deep learning with 3D data
  ☆8,820Updated 2 weeks ago
• pyro-ppl / pyro
  Deep universal probabilistic programming with Python and PyTorch
  ☆8,577Updated 2 weeks ago
• paperswithcode / ai-deadlines
  AI conference deadline countdowns
  ☆5,677Updated 2 months ago
• huggingface / pytorch-image-models
  The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…
  ☆32,290Updated this week
• albumentations-team / albumentations
  Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125
  ☆14,273Updated this week
• snap-stanford / GraphGym
  Platform for designing and evaluating Graph Neural Networks (GNN)
  ☆1,726Updated last year