PyTorch implementation of Variational Diffusion Models.

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

High Fidelity Simulator for Reinforcement Learning and Robotics Research.

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

Training deep learning models on AWS and GCP instances

Template for Jupyterbook Syllabus

Differentiable SDE solvers with GPU support and efficient sensitivity analysis.

Manifold-learning flows (ℳ-flows)

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

Dream to Control: Learning Behaviors by Latent Imagination

Newtonian gravity simulator game that I made 10 years ago

Ladder Variational Autoencoders (LVAE) in PyTorch

Fast and Easy Infinite Neural Networks in Python

Code for experiments in my blog post on the Neural Tangent Kernel: https://eigentales.com/NTK

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

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

A brief computer graphics / rendering course

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

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

repository with the tutorials for MLSS Skoltech

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

POT : Python Optimal Transport

A highly efficient implementation of Gaussian Processes in PyTorch

Geometric loss functions between point clouds, images and volumes

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

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

Differentiable rendering without approximation.