Stars
Train robotic agents to learn to plan pushing and grasping actions for manipulation with deep reinforcement learning.
RoboCasa: Large-Scale Simulation of Everyday Tasks for Generalist Robots
Official repo for VideoComposer: Compositional Video Synthesis with Motion Controllability
A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.
This is an official pytorch implementation of Lite-HRNet: A Lightweight High-Resolution Network.
[ICLR 2024] Official PyTorch implementation of FasterViT: Fast Vision Transformers with Hierarchical Attention
Official Algorithm Implementation of ICML'23 Paper "VIMA: General Robot Manipulation with Multimodal Prompts"
A collection of robotics simulation environments for reinforcement learning
n-dimensional RRT, RRT* (RRT-Star)
LLaVA-Plus: Large Language and Vision Assistants that Plug and Learn to Use Skills
Official code for VisProg (CVPR 2023 Best Paper!)
CALVIN - A benchmark for Language-Conditioned Policy Learning for Long-Horizon Robot Manipulation Tasks
[NeurIPS 2022] Official code for "Focal Modulation Networks"
[ICCV 2023] Official implementation of the paper "A Simple Framework for Open-Vocabulary Segmentation and Detection"
Robot kinematics implemented in pytorch
Access 135+ robot descriptions from the main Python robotics frameworks
A MuJoCo/Gym environment for robot control using Reinforcement Learning. The task of agents in this environment is pixel-wise prediction of grasp success chances.
[CVPR 2022 Oral] Official implementation of DN-DETR
API to run VirtualHome, a Multi-Agent Household Simulator
Implementation of ViViT: A Video Vision Transformer
IKEA Furniture Assembly Environment for Long-Horizon Complex Manipulation Tasks
Perceiver-Actor: A Multi-Task Transformer for Robotic Manipulation
Implementation of RT1 (Robotic Transformer) in Pytorch
Converting Onshape assembly to robot definition (URDF, SDF, MuJoCo) through Onshape API