wohlert

variational-diffusion-models

PyTorch implementation of Variational Diffusion Models.

latex-templates

A collection of LaTeX templates used for research, courses, and miscellanea.

holodeck

High Fidelity Simulator for Reinforcement Learning and Robotics Research.

rrc_simulation

Simulation for the Real Robot Challenge (https://real-robot-challenge.com)

spotty

Training deep learning models on AWS and GCP instances

jupyterbook_course_template

Template for Jupyterbook Syllabus

torchsde

Differentiable SDE solvers with GPU support and efficient sensitivity analysis.

rvae

manifold-flow

Manifold-learning flows (ℳ-flows)

tikzplotlib

:bar_chart: Save matplotlib figures as TikZ/PGFplots for smooth integration into LaTeX.

dreamer

Dream to Control: Learning Behaviors by Latent Imagination

Gravity-Simulator

2D newtonian gravity simulator in JS

pddm

ladder-vae-pytorch

Ladder Variational Autoencoders (LVAE) in PyTorch

neural-tangents

Fast and Easy Infinite Neural Networks in Python

NTK

Code for experiments in my blog post on the Neural Tangent Kernel: https://rajatvd.github.io/NTK

biva-pytorch

Official PyTorch BIVA implementation (BIVA: A Very Deep Hierarchy of Latent Variables for Generative Modeling)

NeurIPS19-SBDRL

Code for NeurIPS 2019 paper: "Symmetry-Based Disentangled Representation Learning requires Interaction with Environments" by H. Caselles-Dupré, M. Garcia-Ortiz and D. Filliat.

tinyrenderer

A brief computer graphics / rendering course

fastapi

FastAPI framework, high performance, easy to learn, fast to code, ready for production

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

tutorials

repository with the tutorials for MLSS Skoltech

pyprob

A probabilistic programming system for simulators and high-performance computing (HPC), based on PyTorch

POT

POT : Python Optimal Transport

gpytorch

A highly efficient implementation of Gaussian Processes in PyTorch

geomloss

Geometric loss functions between point clouds, images and volumes

DiffusionVAE

ProbabilisticNeuralProgrammedNetwork

Code for "Probabilistic Neural Programmed Networks for Scene Generation.", Deng et al, NIPS 2018

torchdiffeq

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

redner

Differentiable rendering without approximation.