Awesome-Pytorch-list

Pytorch & related libraries

pytorch : Tensors and Dynamic neural networks in Python with strong GPU acceleration.

NLP & Speech Processing:

pytorch text : Torch text related contents.
pytorch-seq2seq: A framework for sequence-to-sequence (seq2seq) models implemented in PyTorch.
anuvada: Interpretable Models for NLP using PyTorch.
audio: simple audio I/O for pytorch.
loop: A method to generate speech across multiple speakers
fairseq-py: Facebook AI Research Sequence-to-Sequence Toolkit written in Python.
speech: PyTorch ASR Implementation.
OpenNMT-py: Open-Source Neural Machine Translation in PyTorch http://opennmt.net
neuralcoref: State-of-the-art coreference resolution based on neural nets and spaCy huggingface.co/coref
sentiment-discovery: Unsupervised Language Modeling at scale for robust sentiment classification.
MUSE: A library for Multilingual Unsupervised or Supervised word Embeddings
nmtpytorch: Neural Machine Translation Framework in PyTorch.
pytorch-wavenet: An implementation of WaveNet with fast generation
Tacotron-pytorch: Tacotron: Towards End-to-End Speech Synthesis.
AllenNLP: An open-source NLP research library, built on PyTorch.
PyTorch-NLP: Text utilities and datasets for PyTorch pytorchnlp.readthedocs.io
quick-nlp: Pytorch NLP library based on FastAI.
TTS: Deep learning for Text2Speech
LASER: Language-Agnostic SEntence Representations
pyannote-audio: Neural building blocks for speaker diarization: speech activity detection, speaker change detection, speaker embedding
gensen: Learning General Purpose Distributed Sentence Representations via Large Scale Multi-task Learning.
translate: Translate - a PyTorch Language Library.
espnet: End-to-End Speech Processing Toolkit espnet.github.io/espnet

CV:

pytorch vision : Datasets, Transforms and Models specific to Computer Vision.
pt-styletransfer: Neural style transfer as a class in PyTorch.
OpenFacePytorch: PyTorch module to use OpenFace's nn4.small2.v1.t7 model
img_classification_pk_pytorch: Quickly comparing your image classification models with the state-of-the-art models (such as DenseNet, ResNet, ...)
SparseConvNet: Submanifold sparse convolutional networks.
Convolution_LSTM_pytorch: A multi-layer convolution LSTM module
face-alignment: 🔥 2D and 3D Face alignment library build using pytorch adrianbulat.com
pytorch-semantic-segmentation: PyTorch for Semantic Segmentation.
RoIAlign.pytorch: This is a PyTorch version of RoIAlign. This implementation is based on crop_and_resize and supports both forward and backward on CPU and GPU.
pytorch-cnn-finetune: Fine-tune pretrained Convolutional Neural Networks with PyTorch.
detectorch: Detectorch - detectron for PyTorch
Augmentor: Image augmentation library in Python for machine learning. http://augmentor.readthedocs.io
s2cnn: This library contains a PyTorch implementation of the SO(3) equivariant CNNs for spherical signals (e.g. omnidirectional cameras, signals on the globe) s

Probabilistic/Generative Libraries:

ptstat: Probabilistic Programming and Statistical Inference in PyTorch
pyro: Deep universal probabilistic programming with Python and PyTorch http://pyro.ai
probtorch: Probabilistic Torch is library for deep generative models that extends PyTorch.
paysage: Unsupervised learning and generative models in python/pytorch.
pyvarinf: Python package facilitating the use of Bayesian Deep Learning methods with Variational Inference for PyTorch.
pyprob: A PyTorch-based library for probabilistic programming and inference compilation.

Other libraries:

pytorch extras : Some extra features for pytorch.
functional zoo : PyTorch, unlike lua torch, has autograd in it's core, so using modular structure of torch.nn modules is not necessary, one can easily allocate needed Variables and write a function that utilizes them, which is sometimes more convenient. This repo contains model definitions in this functional way, with pretrained weights for some models.
torch-sampling : This package provides a set of transforms and data structures for sampling from in-memory or out-of-memory data.
torchcraft-py : Python wrapper for TorchCraft, a bridge between Torch and StarCraft for AI research.
aorun : Aorun intend to be a Keras with PyTorch as backend.
logger : A simple logger for experiments.
PyTorch-docset : PyTorch docset! use with Dash, Zeal, Velocity, or LovelyDocs.
convert_torch_to_pytorch : Convert torch t7 model to pytorch model and source.
pretrained-models.pytorch: The goal of this repo is to help to reproduce research papers results.
pytorch_fft : PyTorch wrapper for FFTs
caffe_to_torch_to_pytorch
pytorch-extension: This is a CUDA extension for PyTorch which computes the Hadamard product of two tensors.
tensorboard-pytorch: This module saves PyTorch tensors in tensorboard format for inspection. Currently supports scalar, image, audio, histogram features in tensorboard.
gpytorch: GPyTorch is a Gaussian Process library, implemented using PyTorch. It is designed for creating flexible and modular Gaussian Process models with ease, so that you don't have to be an expert to use GPs.
spotlight: Deep recommender models using PyTorch.
pytorch-cns: Compressed Network Search with PyTorch
pyinn: CuPy fused PyTorch neural networks ops
inferno: A utility library around PyTorch
pytorch-fitmodule: Super simple fit method for PyTorch modules
inferno-sklearn: A scikit-learn compatible neural network library that wraps pytorch.
pytorch-caffe-darknet-convert: convert between pytorch, caffe prototxt/weights and darknet cfg/weights
pytorch2caffe: Convert PyTorch model to Caffemodel
pytorch-tools: Tools for PyTorch
sru: Training RNNs as Fast as CNNs (arxiv.org/abs/1709.02755)
torch2coreml: Torch7 -> CoreML
PyTorch-Encoding: PyTorch Deep Texture Encoding Network http://hangzh.com/PyTorch-Encoding
pytorch-ctc: PyTorch-CTC is an implementation of CTC (Connectionist Temporal Classification) beam search decoding for PyTorch. C++ code borrowed liberally from TensorFlow with some improvements to increase flexibility.
candlegp: Gaussian Processes in Pytorch.
dpwa: Distributed Learning by Pair-Wise Averaging.
dni-pytorch: Decoupled Neural Interfaces using Synthetic Gradients for PyTorch.
skorch: A scikit-learn compatible neural network library that wraps pytorch
ignite: Ignite is a high-level library to help with training neural networks in PyTorch.
Arnold: Arnold - DOOM Agent
pytorch-mcn: Convert models from MatConvNet to PyTorch
simple-faster-rcnn-pytorch: A simplified implemention of Faster R-CNN with competitive performance.
generative_zoo: generative_zoo is a repository that provides working implementations of some generative models in PyTorch.
pytorchviz: A small package to create visualizations of PyTorch execution graphs.
cogitare: Cogitare - A Modern, Fast, and Modular Deep Learning and Machine Learning framework in Python.
pydlt: PyTorch based Deep Learning Toolbox
semi-supervised-pytorch: Implementations of different VAE-based semi-supervised and generative models in PyTorch.
pytorch_cluster: PyTorch Extension Library of Optimised Graph Cluster Algorithms.
neural-assembly-compiler: A neural assembly compiler for pyTorch based on adaptive-neural-compilation.
caffemodel2pytorch: Convert Caffe models to PyTorch.
extension-cpp: C++ extensions in PyTorch
pytoune: A Keras-like framework and utilities for PyTorch
jetson-reinforcement: Deep reinforcement learning libraries for NVIDIA Jetson TX1/TX2 with PyTorch, OpenAI Gym, and Gazebo robotics simulator.
matchbox: Write PyTorch code at the level of individual examples, then run it efficiently on minibatches.
torch-two-sample: A PyTorch library for two-sample tests
pytorch-summary: Model summary in PyTorch similar to model.summary() in Keras
mpl.pytorch: Pytorch implementation of MaxPoolingLoss.
scVI-dev: Development branch of the scVI project in PyTorch
apex: An Experimental PyTorch Extension(will be deprecated at a later point)
ELF: ELF: a platform for game research.
Torchlite: A high level library on top of(not only) Pytorch
joint-vae: Pytorch implementation of JointVAE, a framework for disentangling continuous and discrete factors of variation star2
SLM-Lab: Modular Deep Reinforcement Learning framework in PyTorch.
bindsnet: A Python package used for simulating spiking neural networks (SNNs) on CPUs or GPUs using PyTorch
pro_gan_pytorch: ProGAN package implemented as an extension of PyTorch nn.Module
pytorch_geometric: Geometric Deep Learning Extension Library for PyTorch
torchplus: Implements the + operator on PyTorch modules, returning sequences.
lagom: lagom: A light PyTorch infrastructure to quickly prototype reinforcement learning algorithms.
torchbearer: torchbearer: A model training library for researchers using PyTorch.
pytorch-maml-rl: Reinforcement Learning with Model-Agnostic Meta-Learning in Pytorch.

Tutorials & examples

Paper implementations

google_evolution : This implements one of result networks from Large-scale evolution of image classifiers by Esteban Real, et. al.
pyscatwave : Fast Scattering Transform with CuPy/PyTorch,read the paper here
scalingscattering : Scaling The Scattering Transform : Deep Hybrid Networks.
deep-auto-punctuation : a pytorch implementation of auto-punctuation learned character by character.
Realtime_Multi-Person_Pose_Estimation : This is a pytorch version of Realtime_Multi-Person_Pose_Estimation, origin code is here .
PyTorch-value-iteration-networks : PyTorch implementation of the Value Iteration Networks (NIPS '16) paper
pytorch_Highway : Highway network implemented in pytorch.
pytorch_NEG_loss : NEG loss implemented in pytorch.
pytorch_RVAE : Recurrent Variational Autoencoder that generates sequential data implemented in pytorch.
pytorch_TDNN : Time Delayed NN implemented in pytorch.
eve.pytorch : An implementation of Eve Optimizer, proposed in Imploving Stochastic Gradient Descent with Feedback, Koushik and Hayashi, 2016.
e2e-model-learning : Task-based end-to-end model learning.
pix2pix-pytorch : PyTorch implementation of "Image-to-Image Translation Using Conditional Adversarial Networks".
Single Shot MultiBox Detector : A PyTorch Implementation of Single Shot MultiBox Detector.
DiscoGAN: PyTorch implementation of "Learning to Discover Cross-Domain Relations with Generative Adversarial Networks"
official DiscoGAN implementation : Official implementation of "Learning to Discover Cross-Domain Relations with Generative Adversarial Networks".
pytorch-es : This is a PyTorch implementation of Evolution Strategies .
piwise : Pixel-wise segmentation on VOC2012 dataset using pytorch.
pytorch-dqn : Deep Q-Learning Network in pytorch.
neuraltalk2-pytorch : image captioning model in pytorch(finetunable cnn in branch with_finetune)
vnet.pytorch : A Pytorch implementation for V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation.
pytorch-fcn : PyTorch implementation of Fully Convolutional Networks.
WideResNets : WideResNets for CIFAR10/100 implemented in PyTorch. This implementation requires less GPU memory than what is required by the official Torch implementation: https://github.com/szagoruyko/wide-residual-networks .
pytorch_highway_networks : Highway networks implemented in PyTorch.
pytorch-NeuCom : Pytorch implementation of DeepMind's differentiable neural computer paper.
captionGen : Generate captions for an image using PyTorch.
AnimeGAN : A simple PyTorch Implementation of Generative Adversarial Networks, focusing on anime face drawing.
Cnn-text classification : This is the implementation of Kim's Convolutional Neural Networks for Sentence Classification paper in PyTorch.
deepspeech2 : Implementation of DeepSpeech2 using Baidu Warp-CTC. Creates a network based on the DeepSpeech2 architecture, trained with the CTC activation function.
seq2seq : This repository contains implementations of Sequence to Sequence (Seq2Seq) models in PyTorch
Asynchronous Advantage Actor-Critic in PyTorch : This is PyTorch implementation of A3C as described in Asynchronous Methods for Deep Reinforcement Learning. Since PyTorch has a easy method to control shared memory within multiprocess, we can easily implement asynchronous method like A3C.
densenet : This is a PyTorch implementation of the DenseNet-BC architecture as described in the paper Densely Connected Convolutional Networks by G. Huang, Z. Liu, K. Weinberger, and L. van der Maaten. This implementation gets a CIFAR-10+ error rate of 4.77 with a 100-layer DenseNet-BC with a growth rate of 12. Their official implementation and links to many other third-party implementations are available in the liuzhuang13/DenseNet repo on GitHub.
nninit : Weight initialization schemes for PyTorch nn.Modules. This is a port of the popular nninit for Torch7 by @kaixhin.
faster rcnn : This is a PyTorch implementation of Faster RCNN. This project is mainly based on py-faster-rcnn and TFFRCNN.For details about R-CNN please refer to the paper Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks by Shaoqing Ren, Kaiming He, Ross Girshick, Jian Sun.
doomnet : PyTorch's version of Doom-net implementing some RL models in ViZDoom environment.
flownet : Pytorch implementation of FlowNet by Dosovitskiy et al.
sqeezenet : Implementation of Squeezenet in pytorch, #### pretrained models on CIFAR10 data to come Plan to train the model on cifar 10 and add block connections too.
WassersteinGAN: wassersteinGAN in pytorch.
optnet : This repository is by Brandon Amos and J. Zico Kolter and contains the PyTorch source code to reproduce the experiments in our paper OptNet: Differentiable Optimization as a Layer in Neural Networks.
qp solver : A fast and differentiable QP solver for PyTorch. Crafted by Brandon Amos and J. Zico Kolter.
Continuous Deep Q-Learning with Model-based Acceleration : Reimplementation of Continuous Deep Q-Learning with Model-based Acceleration.
Learning to learn by gradient descent by gradient descent : PyTorch implementation of Learning to learn by gradient descent by gradient descent.
fast-neural-style : pytorch implementation of fast-neural-style, The model uses the method described in Perceptual Losses for Real-Time Style Transfer and Super-Resolution along with Instance Normalization.
PytorchNeuralStyleTransfer : Implementation of Neural Style Transfer in Pytorch.
Fast Neural Style for Image Style Transform by Pytorch : Fast Neural Style for Image Style Transform by Pytorch .
neural style transfer : An introduction to PyTorch through the Neural-Style algorithm (https://arxiv.org/abs/1508.06576) developed by Leon A. Gatys, Alexander S. Ecker and Matthias Bethge.
VIN_PyTorch_Visdom : PyTorch implementation of Value Iteration Networks (VIN): Clean, Simple and Modular. Visualization in Visdom.
YOLO2: YOLOv2 in PyTorch.
attention-transfer: Attention transfer in pytorch, read the paper here.
SVHNClassifier: A PyTorch implementation of Multi-digit Number Recognition from Street View Imagery using Deep Convolutional Neural Networks.
pytorch-deform-conv: PyTorch implementation of Deformable Convolution.
BEGAN-pytorch: PyTorch implementation of BEGAN: Boundary Equilibrium Generative Adversarial Networks.
treelstm.pytorch : Tree LSTM implementation in PyTorch.
AGE: Code for paper "Adversarial Generator-Encoder Networks" by Dmitry Ulyanov, Andrea Vedaldi and Victor Lempitsky which can be found here
ResNeXt.pytorch: Reproduces ResNet-V3 (Aggregated Residual Transformations for Deep Neural Networks) with pytorch.
pytorch-rl: Deep Reinforcement Learning with pytorch & visdom
Deep-Leafsnap: LeafSnap replicated using deep neural networks to test accuracy compared to traditional computer vision methods.
pytorch-CycleGAN-and-pix2pix: PyTorch implementation for both unpaired and paired image-to-image translation.
A3C-PyTorch:PyTorch implementation of Advantage async actor-critic Algorithms (A3C) in PyTorch
pytorch-value-iteration-networks : Pytorch implementation of Value Iteration Networks (NIPS 2016 best paper)
PyTorch-Style-Transfer: PyTorch Implementation of Multi-style Generative Network for Real-time Transfer
pytorch-deeplab-resnet: pytorch-deeplab-resnet-model.
pointnet.pytorch: pytorch implementation for "PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation" https://arxiv.org/abs/1612.00593
pytorch-playground: Base pretrained models and datasets in pytorch (MNIST, SVHN, CIFAR10, CIFAR100, STL10, AlexNet, VGG16, VGG19, ResNet, Inception, SqueezeNet).
pytorch-dnc: Neural Turing Machine (NTM) & Differentiable Neural Computer (DNC) with pytorch & visdom.
pytorch_image_classifier: Minimal But Practical Image Classifier Pipline Using Pytorch, Finetune on ResNet18, Got 99% Accuracy on Own Small Datasets.
mnist-svhn-transfer: PyTorch Implementation of CycleGAN and SGAN for Domain Transfer (Minimal).
pytorch-yolo2: pytorch-yolo2
dni: Implement Decoupled Neural Interfaces using Synthetic Gradients in Pytorch
wgan-gp: A pytorch implementation of Paper "Improved Training of Wasserstein GANs".
pytorch-seq2seq-intent-parsing: Intent parsing and slot filling in PyTorch with seq2seq + attention
pyTorch_NCE: An implementation of the Noise Contrastive Estimation algorithm for pyTorch. Working, yet not very efficient.
molencoder: Molecular AutoEncoder in PyTorch
GAN-weight-norm: Code for "On the Effects of Batch and Weight Normalization in Generative Adversarial Networks"
lgamma: Implementations of polygamma, lgamma, and beta functions for PyTorch
bigBatch : Code used to generate the results appearing in "Train longer, generalize better: closing the generalization gap in large batch training of neural networks"
rl_a3c_pytorch: Reinforcement learning with implementation of A3C LSTM for Atari 2600.
pytorch-retraining: Transfer Learning Shootout for PyTorch's model zoo (torchvision)
nmp_qc: Neural Message Passing for Computer Vision
grad-cam: Pytorch implementation of Grad-CAM
pytorch-trpo: PyTorch Implementation of Trust Region Policy Optimization (TRPO)
pytorch-explain-black-box: PyTorch implementation of Interpretable Explanations of Black Boxes by Meaningful Perturbation
vae_vpflows: Code in PyTorch for the convex combination linear IAF and the Householder Flow, J.M. Tomczak & M. Welling https://jmtomczak.github.io/deebmed.html
relational-networks: Pytorch implementation of "A simple neural network module for relational reasoning" (Relational Networks) https://arxiv.org/pdf/1706.01427.pdf
vqa.pytorch: Visual Question Answering in Pytorch
end-to-end-negotiator: Deal or No Deal? End-to-End Learning for Negotiation Dialogues
odin-pytorch: Principled Detection of Out-of-Distribution Examples in Neural Networks.
FreezeOut: Accelerate Neural Net Training by Progressively Freezing Layers.
ARAE: Code for the paper "Adversarially Regularized Autoencoders for Generating Discrete Structures" by Zhao, Kim, Zhang, Rush and LeCun.
forward-thinking-pytorch: Pytorch implementation of "Forward Thinking: Building and Training Neural Networks One Layer at a Time" https://arxiv.org/pdf/1706.02480.pdf
context_encoder_pytorch: PyTorch Implement of Context Encoders
attention-is-all-you-need-pytorch: A PyTorch implementation of the Transformer model in "Attention is All You Need".https://github.com/thnkim/OpenFacePytorch
OpenFacePytorch: PyTorch module to use OpenFace's nn4.small2.v1.t7 model
neural-combinatorial-rl-pytorch: PyTorch implementation of Neural Combinatorial Optimization with Reinforcement Learning. 95.pytorch-nec: PyTorch Implementation of Neural Episodic Control (NEC)
seq2seq.pytorch: Sequence-to-Sequence learning using PyTorch
Pytorch-Sketch-RNN: a pytorch implementation of arxiv.org/abs/1704.03477
pytorch-pruning: PyTorch Implementation of [1611.06440] Pruning Convolutional Neural Networks for Resource Efficient Inference
DrQA : A pytorch implementation of Reading Wikipedia to Answer Open-Domain Questions.
YellowFin_Pytorch : auto-tuning momentum SGD optimizer
samplernn-pytorch : PyTorch implementation of SampleRNN: An Unconditional End-to-End Neural Audio Generation Model.
AEGeAN: Deeper DCGAN with AE stabilization
/pytorch-SRResNet: pytorch implementation for Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network arXiv:1609.04802v2
vsepp: Code for the paper "VSE++: Improved Visual Semantic Embeddings"
Pytorch-DPPO: Pytorch implementation of Distributed Proximal Policy Optimization: arxiv.org/abs/1707.02286
UNIT: PyTorch Implementation of our Coupled VAE-GAN algorithm for Unsupervised Image-to-Image Translation
efficient_densenet_pytorch: A memory-efficient implementation of DenseNets
tsn-pytorch: Temporal Segment Networks (TSN) in PyTorch.
SMASH: An experimental technique for efficiently exploring neural architectures.
pytorch-retinanet: RetinaNet in PyTorch
biogans: Implementation supporting the ICCV 2017 paper "GANs for Biological Image Synthesis".
Semantic Image Synthesis via Adversarial Learning: A PyTorch implementation of the paper "Semantic Image Synthesis via Adversarial Learning" in ICCV 2017.
fmpytorch: A PyTorch implementation of a Factorization Machine module in cython.
ORN: A PyTorch implementation of the paper "Oriented Response Networks" in CVPR 2017.
pytorch-maml: PyTorch implementation of MAML: arxiv.org/abs/1703.03400
pytorch-generative-model-collections: Collection of generative models in Pytorch version.
vqa-winner-cvprw-2017: Pytorch Implementation of winner from VQA Chllange Workshop in CVPR'17.
tacotron_pytorch : PyTorch implementation of Tacotron speech synthesis model.
pspnet-pytorch: PyTorch implementation of PSPNet segmentation network
LM-LSTM-CRF: Empower Sequence Labeling with Task-Aware Language Model http://arxiv.org/abs/1709.04109
face-alignment: Pytorch implementation of the paper "How far are we from solving the 2D & 3D Face Alignment problem? (and a dataset of 230,000 3D facial landmarks)", ICCV 2017
DepthNet: PyTorch DepthNet Training on Still Box dataset.
EDSR-PyTorch: PyTorch version of the paper 'Enhanced Deep Residual Networks for Single Image Super-Resolution' (CVPRW 2017)
e2c-pytorch: Embed to Control implementation in PyTorch.
3D-ResNets-PyTorch: 3D ResNets for Action Recognition.
bandit-nmt: This is code repo for our EMNLP 2017 paper "Reinforcement Learning for Bandit Neural Machine Translation with Simulated Human Feedback", which implements the A2C algorithm on top of a neural encoder-decoder model and benchmarks the combination under simulated noisy rewards.
pytorch-a2c-ppo-acktr: PyTorch implementation of Advantage Actor Critic (A2C), Proximal Policy Optimization (PPO) and Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation (ACKTR).
zalando-pytorch: Various experiments on the Fashion-MNIST dataset from Zalando.
sphereface_pytorch: A PyTorch Implementation of SphereFace.
Categorical DQN: A PyTorch Implementation of Categorical DQN from A Distributional Perspective on Reinforcement Learning.
pytorch-ntm: pytorch ntm implementation.
mask_rcnn_pytorch: Mask RCNN in PyTorch.
graph_convnets_pytorch: PyTorch implementation of graph ConvNets, NIPS’16
pytorch-faster-rcnn: A pytorch implementation of faster RCNN detection framework based on Xinlei Chen's tf-faster-rcnn.
torchMoji: A pyTorch implementation of the DeepMoji model: state-of-the-art deep learning model for analyzing sentiment, emotion, sarcasm etc.
semantic-segmentation-pytorch: Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset
pytorch-qrnn: PyTorch implementation of the Quasi-Recurrent Neural Network - up to 16 times faster than NVIDIA's cuDNN LSTM
pytorch-sgns: Skipgram Negative Sampling in PyTorch.
SfmLearner-Pytorch : Pytorch version of SfmLearner from Tinghui Zhou et al.
deformable-convolution-pytorch: PyTorch implementation of Deformable Convolution.
skip-gram-pytorch: A complete pytorch implementation of skipgram model (with subsampling and negative sampling). The embedding result is tested with Spearman's rank correlation.
stackGAN-v2: Pytorch implementation for reproducing StackGAN_v2 results in the paper StackGAN++: Realistic Image Synthesis with Stacked Generative Adversarial Networks by Han Zhang*, Tao Xu*, Hongsheng Li, Shaoting Zhang, Xiaogang Wang, Xiaolei Huang, Dimitris Metaxas.
self-critical.pytorch: Unofficial pytorch implementation for Self-critical Sequence Training for Image Captioning.
pygcn: Graph Convolutional Networks in PyTorch.
dnc: Differentiable Neural Computers, for Pytorch
prog_gans_pytorch_inference: PyTorch inference for "Progressive Growing of GANs" with CelebA snapshot.
pytorch-capsule: Pytorch implementation of Hinton's Dynamic Routing Between Capsules.
PyramidNet-PyTorch: A PyTorch implementation for PyramidNets (Deep Pyramidal Residual Networks, arxiv.org/abs/1610.02915)
radio-transformer-networks: A PyTorch implementation of Radio Transformer Networks from the paper "An Introduction to Deep Learning for the Physical Layer". arxiv.org/abs/1702.00832
honk: PyTorch reimplementation of Google's TensorFlow CNNs for keyword spotting.
DeepCORAL: A PyTorch implementation of 'Deep CORAL: Correlation Alignment for Deep Domain Adaptation.', ECCV 2016
pytorch-pose: A PyTorch toolkit for 2D Human Pose Estimation.
lang-emerge-parlai: Implementation of EMNLP 2017 Paper "Natural Language Does Not Emerge 'Naturally' in Multi-Agent Dialog" using PyTorch and ParlAI
Rainbow: Rainbow: Combining Improvements in Deep Reinforcement Learning
pytorch_compact_bilinear_pooling v1: This repository has a pure Python implementation of Compact Bilinear Pooling and Count Sketch for PyTorch.
CompactBilinearPooling-Pytorch v2: (Yang Gao, et al.) A Pytorch Implementation for Compact Bilinear Pooling.
FewShotLearning: Pytorch implementation of the paper "Optimization as a Model for Few-Shot Learning"
meProp: Codes for "meProp: Sparsified Back Propagation for Accelerated Deep Learning with Reduced Overfitting".
SFD_pytorch: A PyTorch Implementation of Single Shot Scale-invariant Face Detector.
GradientEpisodicMemory: Continuum Learning with GEM: Gradient Episodic Memory. https://arxiv.org/abs/1706.08840
DeblurGAN: Pytorch implementation of the paper DeblurGAN: Blind Motion Deblurring Using Conditional Adversarial Networks.
StarGAN: StarGAN: Unified Generative Adversarial Networks for Multi-Domain Image-to-Image Tranlsation.
CapsNet-pytorch: PyTorch implementation of NIPS 2017 paper Dynamic Routing Between Capsules.
CondenseNet: CondenseNet: An Efficient DenseNet using Learned Group Convolutions.
deep-image-prior: Image restoration with neural networks but without learning.
deep-head-pose: Deep Learning Head Pose Estimation using PyTorch.
Random-Erasing: This code has the source code for the paper "Random Erasing Data Augmentation".
FaderNetworks: Fader Networks: Manipulating Images by Sliding Attributes - NIPS 2017
FlowNet 2.0: FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks
pix2pixHD: Synthesizing and manipulating 2048x1024 images with conditional GANs tcwang0509.github.io/pix2pixHD
pytorch-smoothgrad: SmoothGrad implementation in PyTorch
RetinaNet: An implementation of RetinaNet in PyTorch.
faster-rcnn.pytorch: This project is a faster faster R-CNN implementation, aimed to accelerating the training of faster R-CNN object detection models.
mixup_pytorch: A PyTorch implementation of the paper Mixup: Beyond Empirical Risk Minimization in PyTorch.
inplace_abn: In-Place Activated BatchNorm for Memory-Optimized Training of DNNs
pytorch-pose-hg-3d: PyTorch implementation for 3D human pose estimation
nmn-pytorch: Neural Module Network for VQA in Pytorch.
bytenet: Pytorch implementation of bytenet from "Neural Machine Translation in Linear Time" paper
bottom-up-attention-vqa: vqa, bottom-up-attention, pytorch
yolo2-pytorch: The YOLOv2 is one of the most popular one-stage object detector. This project adopts PyTorch as the developing framework to increase productivity, and utilize ONNX to convert models into Caffe 2 to benifit engineering deployment.
reseg-pytorch: PyTorch Implementation of ReSeg (arxiv.org/pdf/1511.07053.pdf)
binary-stochastic-neurons: Binary Stochastic Neurons in PyTorch.
pytorch-pose-estimation: PyTorch Implementation of Realtime Multi-Person Pose Estimation project.
interaction_network_pytorch: Pytorch Implementation of Interaction Networks for Learning about Objects, Relations and Physics.
NoisyNaturalGradient: Pytorch Implementation of paper "Noisy Natural Gradient as Variational Inference".
ewc.pytorch: An implementation of Elastic Weight Consolidation (EWC), proposed in James Kirkpatrick et al. Overcoming catastrophic forgetting in neural networks 2016(10.1073/pnas.1611835114).
pytorch-zssr: PyTorch implementation of 1712.06087 "Zero-Shot" Super-Resolution using Deep Internal Learning
deep_image_prior: An implementation of image reconstruction methods from Deep Image Prior (Ulyanov et al., 2017) in PyTorch.
pytorch-transformer: pytorch implementation of Attention is all you need.
DeepRL-Grounding: This is a PyTorch implementation of the AAAI-18 paper Gated-Attention Architectures for Task-Oriented Language Grounding
deep-forecast-pytorch: Wind Speed Prediction using LSTMs in PyTorch (arxiv.org/pdf/1707.08110.pdf)
cat-net: Canonical Appearance Transformations
minimal_glo: Minimal PyTorch implementation of Generative Latent Optimization from the paper "Optimizing the Latent Space of Generative Networks"
LearningToCompare-Pytorch: Pytorch Implementation for Paper: Learning to Compare: Relation Network for Few-Shot Learning.
poincare-embeddings: PyTorch implementation of the NIPS-17 paper "Poincaré Embeddings for Learning Hierarchical Representations".
pytorch-trpo(Hessian-vector product version): This is a PyTorch implementation of "Trust Region Policy Optimization (TRPO)" with exact Hessian-vector product instead of finite differences approximation.
ggnn.pytorch: A PyTorch Implementation of Gated Graph Sequence Neural Networks (GGNN).
visual-interaction-networks-pytorch: This's an implementation of deepmind Visual Interaction Networks paper using pytorch
adversarial-patch: PyTorch implementation of adversarial patch.
Prototypical-Networks-for-Few-shot-Learning-PyTorch: Implementation of Prototypical Networks for Few Shot Learning (arxiv.org/abs/1703.05175) in Pytorch
Visual-Feature-Attribution-Using-Wasserstein-GANs-Pytorch: Implementation of Visual Feature Attribution using Wasserstein GANs (arxiv.org/abs/1711.08998) in PyTorch.
PhotographicImageSynthesiswithCascadedRefinementNetworks-Pytorch: Photographic Image Synthesis with Cascaded Refinement Networks - Pytorch Implementation
ENAS-pytorch: PyTorch implementation of "Efficient Neural Architecture Search via Parameters Sharing".
Neural-IMage-Assessment: A PyTorch Implementation of Neural IMage Assessment.
proxprop: Proximal Backpropagation - a neural network training algorithm that takes implicit instead of explicit gradient steps.
FastPhotoStyle: A Closed-form Solution to Photorealistic Image Stylization
Deep-Image-Analogy-PyTorch: A python implementation of Deep-Image-Analogy based on pytorch.
Person-reID_pytorch: PyTorch for Person re-ID.
pt-dilate-rnn: Dilated RNNs in pytorch.
pytorch-i-revnet: Pytorch implementation of i-RevNets.
OrthNet: TensorFlow and PyTorch layers for generating Orthogonal Polynomials.
DRRN-pytorch: An implementation of Deep Recursive Residual Network for Super Resolution (DRRN), CVPR 2017
shampoo.pytorch: An implementation of shampoo.
Neural-IMage-Assessment 2: A PyTorch Implementation of Neural IMage Assessment.
TCN: Sequence modeling benchmarks and temporal convolutional networks locuslab/TCN
DCC: This repository contains the source code and data for reproducing results of Deep Continuous Clustering paper.
packnet: Code for PackNet: Adding Multiple Tasks to a Single Network by Iterative Pruning arxiv.org/abs/1711.05769
PyTorch-progressive_growing_of_gans: PyTorch implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation.
nonauto-nmt: PyTorch Implementation of "Non-Autoregressive Neural Machine Translation"
PyTorch-GAN: PyTorch implementations of Generative Adversarial Networks.
PyTorchWavelets: PyTorch implementation of the wavelet analysis found in Torrence and Compo (1998)
pytorch-made: MADE (Masked Autoencoder Density Estimation) implementation in PyTorch
VRNN: Pytorch implementation of the Variational RNN (VRNN), from A Recurrent Latent Variable Model for Sequential Data.
flow: Pytorch implementation of ICLR 2018 paper Deep Learning for Physical Processes: Integrating Prior Scientific Knowledge.
deepvoice3_pytorch: PyTorch implementation of convolutional networks-based text-to-speech synthesis models
psmm: imlementation of the the Pointer Sentinel Mixture Model, as described in the paper by Stephen Merity et al.
tacotron2: Tacotron 2 - PyTorch implementation with faster-than-realtime inference.
AccSGD: Implements pytorch code for the Accelerated SGD algorithm.
QANet-pytorch: an implementation of QANet with PyTorch (EM/F1 = 70.5/77.2 after 20 epoches for about 20 hours on one 1080Ti card.)
ConvE: Convolutional 2D Knowledge Graph Embeddings
Structured-Self-Attention: Implementation for the paper A Structured Self-Attentive Sentence Embedding, which is published in ICLR 2017: arxiv.org/abs/1703.03130 .
graphsage-simple: Simple reference implementation of GraphSAGE.
Detectron.pytorch: A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.
R2Plus1D-PyTorch: PyTorch implementation of the R2Plus1D convolution based ResNet architecture described in the paper "A Closer Look at Spatiotemporal Convolutions for Action Recognition"
StackNN: A PyTorch implementation of differentiable stacks for use in neural networks.
translagent: Code for Emergent Translation in Multi-Agent Communication.
ban-vqa: Bilinear attention networks for visual question answering.
pytorch-openai-transformer-lm: This is a PyTorch implementation of the TensorFlow code provided with OpenAI's paper "Improving Language Understanding by Generative Pre-Training" by Alec Radford, Karthik Narasimhan, Tim Salimans and Ilya Sutskever.
T2F: Text-to-Face generation using Deep Learning. This project combines two of the recent architectures StackGAN and ProGAN for synthesizing faces from textual descriptions.
pytorch - fid: A Port of Fréchet Inception Distance (FID score) to PyTorch
vae_vpflows:Code in PyTorch for the convex combination linear IAF and the Householder Flow, J.M. Tomczak & M. Welling jmtomczak.github.io/deebmed.html
CoordConv-pytorch: Pytorch implementation of CoordConv introduced in 'An intriguing failing of convolutional neural networks and the CoordConv solution' paper. (arxiv.org/pdf/1807.03247.pdf)
SDPoint: Implementation of "Stochastic Downsampling for Cost-Adjustable Inference and Improved Regularization in Convolutional Networks", published in CVPR 2018.
SRDenseNet-pytorch: SRDenseNet-pytorch（ICCV_2017）
GAN_stability: Code for paper "Which Training Methods for GANs do actually Converge? (ICML 2018)"
Mask-RCNN: A PyTorch implementation of the architecture of Mask RCNN, serves as an introduction to working with PyTorch
pytorch-coviar: Compressed Video Action Recognition
PNASNet.pytorch: PyTorch implementation of PNASNet-5 on ImageNet.

Pytorch elsewhere

the-incredible-pytorch : The Incredible PyTorch: a curated list of tutorials, papers, projects, communities and more relating to PyTorch.
generative models : Collection of generative models, e.g. GAN, VAE in Tensorflow, Keras, and Pytorch. http://wiseodd.github.io
pytorch vs tensorflow : an informative thread on reddit.
Pytorch discussion forum
pytorch notebook: docker-stack : A project similar to Jupyter Notebook Scientific Python Stack
drawlikebobross: Draw like Bob Ross using the power of Neural Networks (With PyTorch)!
pytorch-tvmisc: Totally Versatile Miscellanea for Pytorch
pytorch-a3c-mujoco: Implement A3C for Mujoco gym envs.
PyTorch in 5 Minutes.
pytorch_chatbot: A Marvelous ChatBot implemented using PyTorch.
malmo-challenge: Malmo Collaborative AI Challenge - Team Pig Catcher
sketchnet: A model that takes an image and generates Processing source code to regenerate that image
Deep-Learning-Boot-Camp: A nonprofit community run, 5-day Deep Learning Bootcamp http://deep-ml.com.
Amazon_Forest_Computer_Vision: Satellite Image tagging code using PyTorch / Keras with lots of PyTorch tricks. kaggle competition.
AlphaZero_Gomoku: An implementation of the AlphaZero algorithm for Gomoku (also called Gobang or Five in a Row)
pytorch-cv: Repo for Object Detection, Segmentation & Pose Estimation.
deep-person-reid: Pytorch implementation of deep person re-identification approaches.
pytorch-template: PyTorch template project
Deep Learning With Pytorch TextBook A practical guide to build neural network models in text and vision using PyTorch. Purchase on Amazon github code repo
compare-tensorflow-pytorch: Compare outputs between layers written in Tensorflow and layers written in Pytorch.

Feedback: If you have any ideas or you want any other content to be added to this list, feel free to contribute.

stephenivy07/Awesome-pytorch-list

ErrorLooks like something went wrong!

About

Resources

Uh oh!

Stars

Watchers

Forks

Releases

Packages 0

Error
Looks like something went wrong!

Packages