timgaripov

raylib

A simple and easy-to-use library to enjoy videogames programming

vision

Datasets, Transforms and Models specific to Computer Vision

deep-image-prior

Image restoration with neural networks but without learning.

torchdiffeq

Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.

drake

Model-based design and verification for robotics.

ffcv

FFCV: Fast Forward Computer Vision (and other ML workloads!)

loss-landscape

Code for visualizing the loss landscape of neural nets

tsne-cuda

GPU Accelerated t-SNE for CUDA with Python bindings

pytorch-classification

Classification with PyTorch.

DomainBed

DomainBed is a suite to test domain generalization algorithms

RelativisticGAN

Code for replication of the paper "The relativistic discriminator: a key element missing from standard GAN"

mimicry

[CVPR 2020 Workshop] A PyTorch GAN library that reproduces research results for popular GANs.

WeightStandardization

Standardizing weights to accelerate micro-batch training

wide-resnet.pytorch

Best CIFAR-10, CIFAR-100 results with wide-residual networks using PyTorch

swa_gaussian

Code repo for "A Simple Baseline for Bayesian Uncertainty in Deep Learning"

LieConv

Diffusion-Policies-for-Offline-RL

understandingbdl

t3f

Tensor Train decomposition on TensorFlow

fastswa-semi-sup

Improving Consistency-Based Semi-Supervised Learning with Weight Averaging

deep_feature_reweighting

rl_with_resets

JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"

MMD-GAN

Improving MMD-GAN training with repulsive loss function

adagan

AdaGAN: greedy iterative procedure to train mixtures of GANs

fullbatchtraining

Training vision models with full-batch gradient descent and regularization

mujoco_tips_and_tricks

PairGAN

asa

Code for paper "Adversarial Support Alignment"

jaynes

conveyor-detection

"Deep Hough Transform for Semantic Line Detection" (ECCV 2020, PAMI 2021) adaptation for the conveyor detection problem