pyg-team / pytorch_geometric

Related projects: ⓘ

• dmlc / dgl
  Python package built to ease deep learning on graph, on top of existing DL frameworks.
  ☆13,361Updated this week
• thunlp / GNNPapers
  Must-read papers on graph neural networks (GNN)
  ☆15,871Updated 8 months ago
• HarisIqbal88 / PlotNeuralNet
  Latex code for making neural networks diagrams
  ☆21,827Updated last year
• tkipf / gcn
  Implementation of Graph Convolutional Networks in TensorFlow
  ☆7,081Updated last year
• pytorch / examples
  A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
  ☆22,236Updated this week
• huggingface / pytorch-image-models
  The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…
  ☆31,479Updated this week
• jadore801120 / attention-is-all-you-need-pytorch
  A PyTorch implementation of the Transformer model in "Attention is All You Need".
  ☆8,748Updated 5 months ago
• tkipf / pygcn
  Graph Convolutional Networks in PyTorch
  ☆5,146Updated 3 years ago
• pytorch / vision
  Datasets, Transforms and Models specific to Computer Vision
  ☆15,992Updated this week
• yunjey / pytorch-tutorial
  PyTorch Tutorial for Deep Learning Researchers
  ☆29,835Updated last year
• naganandy / graph-based-deep-learning-literature
  links to conference publications in graph-based deep learning
  ☆4,742Updated last week
• Lightning-AI / pytorch-lightning
  Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.
  ☆27,963Updated this week
• eriklindernoren / PyTorch-GAN
  PyTorch implementations of Generative Adversarial Networks.
  ☆16,230Updated 3 months ago
• lanpa / tensorboardX
  tensorboard for pytorch (and chainer, mxnet, numpy, ...)
  ☆7,852Updated 2 weeks ago
• albumentations-team / albumentations
  Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125
  ☆14,031Updated this week
• bharathgs / Awesome-pytorch-list
  A comprehensive list of pytorch related content on github,such as different models,implementations,helper libraries,tutorials etc.
  ☆15,313Updated 7 months ago
• NVIDIA / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆8,303Updated 2 weeks ago
• pytorch / tutorials
  PyTorch tutorials.
  ☆8,102Updated this week
• lucidrains / vit-pytorch
  Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Py…
  ☆19,651Updated 3 weeks ago
• arogozhnikov / einops
  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
  ☆8,362Updated this week
• tensorflow / tensor2tensor
  Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.
  ☆15,324Updated last year
• google-research / google-research
  Google Research
  ☆33,807Updated this week
• optuna / optuna
  A hyperparameter optimization framework
  ☆10,527Updated last week
• dair-ai / ml-visuals
  🎨 ML Visuals contains figures and templates which you can reuse and customize to improve your scientific writing.
  ☆13,147Updated last year
• google-deepmind / graph_nets
  Build Graph Nets in Tensorflow
  ☆5,344Updated last year
• NVIDIA / DeepLearningExamples
  State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enter…
  ☆13,233Updated last month
• jason718 / awesome-self-supervised-learning
  A curated list of awesome self-supervised methods
  ☆6,113Updated 2 months ago
• graykode / nlp-tutorial
  Natural Language Processing Tutorial for Deep Learning Researchers
  ☆14,070Updated 6 months ago
• paperswithcode / ai-deadlines
  AI conference deadline countdowns
  ☆5,582Updated this week
• google-research / vision_transformer
  ☆10,072Updated 3 months ago