This is a curated list of tutorials, projects, libraries, videos, papers, books and anything related to the incredible PyTorch. Feel free to make a pull request to contribute to this list.
- Official PyTorch Tutorials
- Official PyTorch Examples
- Practical Deep Learning with PyTorch by Deep Learning Wizard (Ritchie Ng)
- Simple Examples to Introduce PyTorch
- Mini Tutorials in PyTorch
- Deep Learning for NLP
- Deep Learning Tutorial for Researchers
- Fully Convolutional Networks implemented with PyTorch
- Simple PyTorch Tutorials Zero to ALL
- DeepNLP-models-Pytorch
- MILA PyTorch Welcome Tutorials
- Practical PyTorch
- PyTorch Project Template
- A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.
- Get started Tutorial
- Mnist Tutorial
- ERFNET
- DCGAN
- DQN
- Ordered Neurons: Integrating Tree Structures into Recurrent Neural Networks
- Loss Visualization
- PyTorch-Kaldi Speech Recognition Toolkit
- Hierarchical Multi-Task Learning Model
- WaveGlow: A Flow-based Generative Network for Speech Synthesis
- BERT
- Quantile Regression DQN
- Accurate Neural Network Potential
- L-BFGS
- Stochastic Downsampling for Cost-Adjustable Inference and Improved Regularization in Convolutional Networks
- Progressive Growing of GANs for Improved Quality, Stability, and Variation
- PyTorch Realtime Multi-Person Pose Estimation
- Faster Faster R-CNN Implementation
- In-Place Activated BatchNorm for Memory-Optimized Training of DNNs
- Wasserstein GAN
- OptNet: Differentiable Optimization as a Layer in Neural Networks
- Paying More Attention to Attention: Improving the Performance of Convolutional Neural Networks via Attention Transfer
- Wide ResNet model in PyTorch
- Task-based End-to-end Model Learning
- An End-to-End Trainable Neural Network for Image-based Sequence Recognition and Its Application to Scene Text Recognition
- Scaling the Scattering Transform: Deep Hybrid Networks
- Adversarial Generator-Encoder Network
- Conditional Similarity Networks
- Multi-style Generative Network for Real-time Transfer
- Image-to-Image Translation with Conditional Adversarial Networks
- Unpaired Image-to-Image Translation using Cycle-Consistent Adversarial Networks
- Inferring and Executing Programs for Visual Reasoning
- On the Effects of Batch and Weight Normalization in Generative Adversarial Networks
- Train longer, generalize better: closing the generalization gap in large batch training of neural networks
- Neural Message Passing for Quantum Chemistry
- DiracNets: Training Very Deep Neural Networks Without Skip-Connections
- Deal or No Deal? End-to-End Learning for Negotiation Dialogues
- Visual Question Answering in Pytorch
- Principled Detection of Out-of-Distribution Examples in Neural Networks
- Attention is all you need
- FreezeOut: Accelerate Training by Progressively Freezing Layers
- CortexNet: a Generic Network Family for Robust Visual Temporal Representations
- VSE++: Improved Visual-Semantic Embeddings
- Reading Wikipedia to Answer Open-Domain Questions
- A Structured Self-Attentive Sentence Embedding
- Efficient Densenet
- Averaged Stochastic Gradient Descent with Weight Dropped LSTM
- Oriented Response Networks
- Video Frame Interpolation via Adaptive Separable Convolution
- Learning local feature descriptors with triplets and shallow convolutional neural networks
- Training RNNs as Fast as CNNs
- How far are we from solving the 2D & 3D Face Alignment problem? (and a dataset of 230,000 3D facial landmarks)
- Learning Spatio-Temporal Features with 3D Residual Networks for Action Recognition
- Reinforcement Learning for Bandit Neural Machine Translation with Simulated Human Feedback
- SphereFace: Deep Hypersphere Embedding for Face Recognition
- Quasi-Recurrent Neural Network (QRNN)
- Enhanced Deep Residual Networks for Single Image Super-Resolution
- DiracNets
- pytorch-CycleGAN-and-pix2pixpytorch-CycleGAN-and-pix2pix
- OpenNMT
- Superresolution using an efficient sub-pixel convolutional neural network
- Deep Learning for Physical Processes: Integrating Prior Scientific Knowledge
- GANimation: Anatomically-aware Facial Animation from a Single Image (ECCV'18 Oral)
- Associative Compression Networks
- Clarinet
- Continuous Wavelet Transforms
- Neural Sequence labeling model
- Improved Training of Wasserstein GANs
- Binary Stochastic Neurons
- ReSeg: A Recurrent Neural Network-based Model for Semantic Segmentation
- mixup: Beyond Empirical Risk Minimization
- Towards 3D Human Pose Estimation in the Wild: a Weakly-supervised Approach
- Hierarchical Attention Network for Document Classification
- Unsupervised Learning of Depth and Ego-Motion from Video
- Learning to learn by gradient descent by gradient descent
- Densely Connected Convolutional Networks
- A Neural Algorithm of Artistic Style
- Very Deep Convolutional Networks for Large-Scale Image Recognition
- VGG model in PyTorch.
- SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size
- Network In Network
- Deep Residual Learning for Image Recognition
- ResNet model in PyTorch.
- Training Wide ResNets for CIFAR-10 and CIFAR-100 in PyTorch
- Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks
- FlowNet: Learning Optical Flow with Convolutional Networks
- Asynchronous Methods for Deep Reinforcement Learning
- Perceptual Losses for Real-Time Style Transfer and Super-Resolution
- Highway Networks
- Collection of Generative Models with PyTorch
- Generative Adversarial Nets (GAN)
- Variational Autoencoder (VAE)
- A Recurrent Latent Variable Model for Sequential Data (VRNN)
- Hybrid computing using a neural network with dynamic external memory
- Deep Speech 2: End-to-End Speech Recognition in English and Mandarin
- V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation
- Value Iteration Networks
- YOLOv2: Real-Time Object Detection
- Convolutional Neural Fabrics
- Improved Semantic Representations From Tree-Structured Long Short-Term Memory Networks
- PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation
- Deformable Convolutional Network
- Continuous Deep Q-Learning with Model-based Acceleration
- Hierarchical Attention Networks for Document Classification
- Spatial Transformer Networks
- Decoupled Neural Interfaces using Synthetic Gradients
- Improved Training of Wasserstein GANs
- CycleGAN and Semi-Supervised GAN
- Automatic chemical design using a data-driven continuous representation of molecules
- Differentiable Neural Computer
- Asynchronous Methods for Deep Reinforcement Learning for Atari 2600
- Trust Region Policy Optimization
- Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localization
- Improving Variational Auto-Encoders using Householder Flow and using convex combination linear Inverse Autoregressive Flow
- SSD: Single Shot MultiBox Detector
- Neural Combinatorial Optimization with Reinforcement Learning
- Pruning Convolutional Neural Networks for Resource Efficient Inference
- A Neural Representation of Sketch Drawings
- Convolutional LSTM Network
- Noisy Networks for Exploration
- Multi-digit Number Recognition from Street View Imagery using Deep Convolutional Neural Networks
- Distributed Proximal Policy Optimization
- Single Shot MultiBox Detector
- Deformable Convolutional Networks in PyTorch
- Dilated ResNet combination with Dilated Convolutions
- Attention-Based Recurrent Neural Network Models for Joint Intent Detection and Slot Filling
- Attentive Recurrent Comparators
- PyTorch GAN Collection
- Compact Bilinear Pooling
- Striving for Simplicity: The All Convolutional Net
- Deep Inside Convolutional Networks: Visualising Image Classification Models and Saliency Maps
- Deep Neural Networks are Easily Fooled: High Confidence Predictions for Unrecognizable Images
- Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localization - 2
- FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks
- Optical Flow Estimation using a Spatial Pyramid Network
- Skip-Thought Vectors
- Explaining and Harnessing Adversarial Examples
- Understanding Deep Image Representations by Inverting Them
- NIMA: Neural Image Assessment
- NASNet-A-Mobile. Ported weights
- SmoothGrad: removing noise by adding noise
- Interpretable Counting for Visual Question Answering
- Big collection of pretrained classification models
- Delira, lightweight framework for medical imaging prototyping
- PyTorch Inference on C++ in Windows
- Shap, a unified approach to explain the output of any machine learning model
- Mixed Precision Training in PyTorch
- Medical Torch, medical imaging framework for PyTorch
- EuclidesDB, Multi-Model Machine Learning Feature Database
- DeOldify, Coloring Old Images
- Perturbative Neural Networks
- Detectron models for Object Detection
- Collection of Sequence to Sequence Models with PyTorch
- Vanilla Sequence to Sequence models
- Attention based Sequence to Sequence models
- Faster attention mechanisms using dot products between the final encoder and decoder hidden states
- Reinforcement learning models in ViZDoom environment with PyTorch
- Neuraltalk 2, Image Captioning Model, in PyTorch
- Recurrent Neural Networks for Sentiment Analysis (Aspect-Based) on SemEval 2014
- PyTorch Image Classification with Kaggle Dogs vs Cats Dataset
- CNN Based Text Classification
- Open-source (MIT) Neural Machine Translation (NMT) System
- Pytorch Poetry Generation
- Data Augmentation and Sampling for Pytorch
- CIFAR-10 on Pytorch with VGG, ResNet and DenseNet
- Generate captions from an image with PyTorch
- Generative Adversarial Networks, focusing on anime face drawing
- Simple Generative Adversarial Networks
- Fast Neural Style Transfer
- Pixel-wise Segmentation on VOC2012 Dataset using PyTorch
- Draw like Bob Ross
- Reinforcement learning models using Gym and Pytorch
- Open-Source Neural Machine Translation in PyTorch
- Deep Video Analytics
- Adversarial Auto-encoders
- Whale Detector
- Base pretrained models and datasets in pytorch (MNIST, SVHN, CIFAR10, CIFAR100, STL10, AlexNet, VGG16, VGG19, ResNet, Inception, SqueezeNet)
- Open Source Chatbot with PyTorch
- Seq2Seq Intent Parsing
- OpenFace in PyTorch
- Complete Suite for Training Seq2Seq Models in PyTorch
- Probabilistic Programming and Statistical Inference in PyTorch
- Graphics code generating model using Processing
- MUSE: Multilingual Unsupervised and Supervised Embeddings
- SLM-Lab: Modular Deep Reinforcement Learning framework in PyTorch
- torchgan: Framework for modelling Generative Adversarial Networks in Pytorch
- PyText, deep learning based NLP modelling framework officially maintained by FAIR
- Torchstat for Statistics on PyTorch Models
- Load Audio files directly into PyTorch Tensors
- Weight Initializations
- Spatial transformer implemented in PyTorch
- PyTorch AWS AMI, run PyTorch with GPU support in less than 5 minutes
- Use tensorboard with PyTorch
- Simple Fit Module in PyTorch, similar to Keras
- torchbearer: A model fitting library for PyTorch
- PyTorch to Keras model converter
- Gluon to PyTorch model converter with code generation
Do feel free to contribute!
You can raise an issue or submit a pull request, whichever is more convenient for you. The guideline is simple: just follow the format of the previous bullet point.