Stars
Re-implementation of pi0 vision-language-action (VLA) model from Physical Intelligence
Python package to create manipulation scenes.
A Vision-Language Model for Spatial Affordance Prediction in Robotics
M2T2: Multi-Task Masked Transformer for Object-centric Pick and Plac
Octo is a transformer-based robot policy trained on a diverse mix of 800k robot trajectories.
A PyTorch implementation of the CVPR 2022 Paper "Neural RGB-D Surface Reconstruction"
The repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.
Source code and dataset for 2022 Publication "Applying Deep Reinforcement Learning to the HP Model for Protein Structure Prediction"
A Python framework for accelerated simulation, data generation and spatial computing.
This repo has the expert data generation infrastructure and Pytorch implementation of MPiNets.
A library for creating Gym environments with unified API to various physics simulators
A simulation environment and benchmark for human-to-robot object handovers
A template for learning-based projects with managed, reproducible experiments
Code for the paper "Planning with Diffusion for Flexible Behavior Synthesis"
Calculate signed distance fields for arbitrary meshes
PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
Instant neural graphics primitives: lightning fast NeRF and more
Frustum PointNets for 3D Object Detection from RGB-D Data
A library for flexible voxel-based mapping, mainly focusing on truncated and Euclidean signed distance fields.
Hypersim: A Photorealistic Synthetic Dataset for Holistic Indoor Scene Understanding
A modular high-level library to train embodied AI agents across a variety of tasks and environments.
NeRF (Neural Radiance Fields) and NeRF in the Wild using pytorch-lightning