PyTorch GAN - PyTorch implementations of Generative Adversarial Networks.
OCaml PyTorch - OCaml bindings for PyTorch.
Apex - PyTorch Extension: Tools for easy mixed precision and distributed training in PyTorch.
Translate - PyTorch Language Library.
Neural Pipeline - Neural networks training pipeline based on PyTorch. Designed to standardize training process and accelerate experiments.
tch-rs - Rust bindings for PyTorch.
PyTorch Geometric - Geometric deep learning extension library for PyTorch.
BigGAN-PyTorch - Contains code for 4-8 GPU training of BigGANs from Large Scale GAN Training for High Fidelity Natural Image Synthesis.
Incredible PyTorch - Curated list of tutorials, papers, projects, communities and more relating to PyTorch.
NimTorch - Nim frontend for pytorch, aiming to be mostly auto-generated and internally using ATen.
XLM - PyTorch original implementation of Cross-lingual Language Model Pretraining.
BoTorch - Bayesian optimization in PyTorch.
PyTorch Lightning - Rapid research framework for PyTorch. The researcher's version of Keras.
PointNet.pytorch - PyTorch implementation for "PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation".
former - Simple transformer implementation from scratch in PyTorch.
PyTorch BigGraph - Software used for generating embeddings from large-scale graph-structured data.
Deep Learning (PyTorch) - Projects and exercises for the latest Deep Learning ND program.
PyWarm - Cleaner way to build neural networks for PyTorch.
Higher - PyTorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual training steps.
Detectron2 - FAIR's next-generation research platform for object detection and segmentation.
Neural Network Programming - Deep Learning with PyTorch (2019) - Series is all about neural network programming and PyTorch.
Captum - Model interpretability and understanding for PyTorch.
Pytorch-Struct - Fast, general, and tested differentiable structured prediction in PyTorch.
3D Ken Burns Effect - Implementation of 3D Ken Burns Effect from a Single Image using PyTorch.
Kaolin - PyTorch Library for Accelerating 3D Deep Learning Research.
SpConv - PyTorch Spatially Sparse Convolution Library.
Segmentation models - Python library with Neural Networks for Image Segmentation based on PyTorch.
cvxpylayers - Python library for constructing differentiable convex optimization layers in PyTorch and TensorFlow using CVXPY.
Frustum ConvNet - Sliding Frustums to Aggregate Local Point-Wise Features for Amodal 3D Object Detection.
PyTorch Elastic - Framework that enables distributed training jobs to be executed in a fault tolerant and elastic manner.
torchgpipe - GPipe implementation in PyTorch.
CycleGAN and pix2pix in PyTorch - Image-to-Image Translation in PyTorch.
TTAch - Image Test Time Augmentation with PyTorch.
OpenProtein - PyTorch framework for tertiary protein structure prediction.
PyTorch Toolbelt - PyTorch extensions for fast R&D prototyping and Kaggle farming.
Reformer - Efficient Transformer, in PyTorch.
Open Graph Benchmark (OGB) - Collection of benchmark datasets, data-loaders and evaluators for graph machine learning in PyTorch.
Ignite - High-level library to help with training neural networks in PyTorch.
PyTorch Lightning - Lightweight PyTorch wrapper for ML researchers. Scale your models. Write less boilerplate.
face.evoLVe - High-Performance Face Recognition Library based on PyTorch.
Fast AutoAugment - Official Implementation of 'Fast AutoAugment' in PyTorch.
Matchbox - Enables deep learning researchers to write PyTorch code at the level of individual examples, then run it efficiently on minibatches.
torchtext - Data loaders and abstractions for text and NLP.
Relational Networks - PyTorch implementation of "A simple neural network module for relational reasoning".
nflows - Normalizing flows in PyTorch.
torchlayers - Shape inference for PyTorch (like in Keras) + new SoTA layers.
Torchmeta - Collection of extensions and data-loaders for few-shot learning & meta-learning in PyTorch.
PyTorch Scatter - PyTorch Extension Library of Optimized Scatter Operations.
TorchServe - Flexible and easy to use tool for serving PyTorch models.
torch-optimizer - Collection of optimizers for PyTorch.
Tacotron 2 (without wavenet) - PyTorch implementation of Natural TTS Synthesis By Conditioning Wavenet On Mel Spectrogram Predictions.
pytorch-crf - Conditional random field in PyTorch.
torchprof - PyTorch layer-by-layer model profiler.
rectorch - PyTorch-based framework for state-of-the-art top-N recommendation.
neural-dream - PyTorch implementation of DeepDream algorithm.
PyProf2 - PyTorch Profiling tool.
Blitz - Simple and extensible library to create Bayesian Neural Network layers on PyTorch.
PyContrast - PyTorch implementation of Contrastive Learning methods.
PyVacy - Differentially Private Optimization Algorithms for PyTorch.
Effective PyTorch - PyTorch tutorials and best practices.
PyTorch Toolbox - Aiming to make you write Pytorch code more easier, readable and concise.
PyTorch Hub - Discover and publish models to a pre-trained model repository designed for research exploration.
DrQA - Reading Wikipedia to Answer Open-Domain Questions. In PyTorch.