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

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

Code for paper "Adversarial Support Alignment"

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

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

Training vision models with full-batch gradient descent and regularization

Model-based design and verification for robotics.

DomainBed is a suite to test domain generalization algorithms

Improving MMD-GAN training with repulsive loss function

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

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

Standardizing weights to accelerate micro-batch training

GPU Accelerated t-SNE for CUDA with Python bindings

Code for visualizing the loss landscape of neural nets

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

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

Improving Consistency-Based Semi-Supervised Learning with Weight Averaging

Image restoration with neural networks but without learning.

Datasets, Transforms and Models specific to Computer Vision

Classification with PyTorch.

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

Tensor Train decomposition on TensorFlow

AdaGAN: greedy iterative procedure to train mixtures of GANs

Experiment code for Stochastic Gradient Hamiltonian Monte Carlo