[RSS 2023] Diffusion Policy Visuomotor Policy Learning via Action Diffusion

This code corresponds to simulation environments used as part of the MimicGen project.

Computations and statistics on manifolds with geometric structures.

Monte Carlo tree search in JAX

Calculate signed distance fields for arbitrary meshes

Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch

This repository contains the code for the paper "Occupancy Networks - Learning 3D Reconstruction in Function Space"

Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"

Python library for Reinforcement Learning.

Robot kinematics implemented in pytorch

A PyTorch implementation of Implicit Behavioral Cloning by Florence et al.

Animation engine for explanatory math videos

Python tool for obtaining watertight meshes using TSDF fusion.

Model Predictive Actor-Critic Reinforcement Learning

Manifold-modeling flows (MFMFs)

A large-scale benchmark and learning environment.

Code for sequential flows.

Companion code to "Learning Stable Deep Dynamics Models" (Manek and Kolter, 2019)

Preliminary code of paper: Model Based Planning with Energy Based Models

hand-eye calibration, tool-flange calibration