robotgradient

grasp_diffusion

Pytorch implementation of diffusion models on Lie Groups for 6D grasp pose generation https://sites.google.com/view/se3dif/home

iflow

LieFlows

Pytorch implementation of Stable Vector Fields on Lie Groups through Diffeomorphism

mesh_to_sdf

Calculate signed distance fields for arbitrary meshes

geometric_fabrics

training_baseline

DALLE2-pytorch

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

diffusion_policy

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

EB_GFN

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

geomstats

Computations and statistics on manifolds with geometric structures.

ibc

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

manifold-flow

Manifold-modeling flows (MFMFs)

manim

Animation engine for explanatory math videos

mctx

Monte Carlo tree search in JAX

mesh-fusion

Python tool for obtaining watertight meshes using TSDF fusion.

mimicgen_environments

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

model_based_planning_ebm

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

mopac

Model Predictive Actor-Critic Reinforcement Learning

mushroom-rl

Python library for Reinforcement Learning.

occupancy_networks

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

Pose-Estimation-for-Sensor-Calibration

hand-eye calibration, tool-flange calibration

pytorch_kinematics

Robot kinematics implemented in pytorch

RLBench

A large-scale benchmark and learning environment.

robotgradient

robotgradient.github.io

sds

A toolbox for inference of switching systems for control

sequential_flows

Code for sequential flows.

stable_dynamics

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

Trajectories

visuo_language_6dofdiffskills