szagoruyko

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

visdom

A flexible tool for creating, organizing, and sharing visualizations of live, rich data. Supports Torch and Numpy.

sonnet

TensorFlow-based neural network library

OpenNMT-py

Open Source Neural Machine Translation and (Large) Language Models in PyTorch

ssd.pytorch

A PyTorch Implementation of Single Shot MultiBox Detector

pytorch-examples

Simple examples to introduce PyTorch

fairseq-lua

Facebook AI Research Sequence-to-Sequence Toolkit

WassersteinGAN

ResNeXt

Implementation of a classification framework from the paper Aggregated Residual Transformations for Deep Neural Networks

attention-transfer

Improving Convolutional Networks via Attention Transfer (ICLR 2017)

scikit-cuda

Python interface to GPU-powered libraries

crayon

A language-agnostic interface to TensorBoard

tensorboard_logger

Log TensorBoard events without touching TensorFlow

diracnets

Training Very Deep Neural Networks Without Skip-Connections

tikz_cnn

Macro to draw architecture of convolutional neural nets in tikz.

optnet

OptNet: Differentiable Optimization as a Layer in Neural Networks

topicsne

t-SNE experiments in pytorch

ARAE

Code for the paper "Adversarially Regularized Autoencoders (ICML 2018)" by Zhao, Kim, Zhang, Rush and LeCun

pytorch_fft

PyTorch wrapper for FFTs

variational-dropout-sparsifies-dnn

Sparse Variational Dropout, ICML 2017

pyscatwave

Fast Scattering Transform with CuPy/PyTorch

shake-shake

2.86% and 15.85% on CIFAR-10 and CIFAR-100

tensorflow-model-zoo.torch

InceptionV3, InceptionV4, Inception-Resnet pretrained models for Torch7 and PyTorch

deep-smile-warp

DeepWarp for Facial Expression Manipulation

neural-style-audio-torch

Torch implementation for audio neural style.

mlogger

a lightweight and simple logger for Machine Learning

scalingscattering

wide-residual-network

Wide-residual network implementations. Best result for cifar10(97.12%), cifar100(84.12%), and other kaggle challenges

eve.pytorch

DiracNets