IVON optimizer for neural networks based on variational learning.

Coherent Soft Imitation Learning

Implementation of Motion Planning Diffusion

Second-order iterated smoothing algorithms for state estimation

Massively parallel rigidbody physics simulation on accelerator hardware.

Implementation for Proximal Point Imitation Learning

Iterative Linearized Control Toolbox

Solve puzzles. Improve your pytorch.

Cleaner and Formatter for BibTeX files

KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows

Examples of how to create colorful, annotated equations in Latex using Tikz.

Code for "Model-based Reinforcement Learning for Continuous Control with Posterior Sampling", ICML 2021

Reinforcement learning library in JAX.

This repo contains a PyTorch implementation of the paper: "Evidential Deep Learning to Quantify Classification Uncertainty"

Belief matching framework official implementation

Library for Model Based RL

Multi-Joint dynamics with Contact. A general purpose physics simulator.

High-quality implementations of standard and SOTA methods on a variety of tasks.

Efficient Differentiable Simulation of Articulated Bodies (ICML2021)

Sparse nonlinear least squares for JAX

Nimble: Physics Engine for Biomechanics and Deep Learning

Sequential Monte Carlo in python

Extensible Tensorflow library for differentiable particle filtering. ICML 2021.

Implementation of different Relative Entropy Policy Search flavors