kunni918

cleanrl

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

cpl

Code for Contrastive Preference Learning (CPL)

datasets

🤗 The largest hub of ready-to-use datasets for ML models with fast, easy-to-use and efficient data manipulation tools

diffusion_policy

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

DTPP

[ICRA 2024] Differentiable Joint Conditional Prediction and Cost Evaluation for Tree Policy Planning

generative-ai-for-beginners

18 Lessons, Get Started Building with Generative AI 🔗 https://microsoft.github.io/generative-ai-for-beginners/

GRUtopia

GRUtopia: Dream General Robots in a City at Scale

gym-aloha

A gym environment for ALOHA

humanoid-gym

Humanoid-Gym: Reinforcement Learning for Humanoid Robot with Zero-Shot Sim2Real Transfer https://arxiv.org/abs/2404.05695

humanplus

HumanPlus: Humanoid Shadowing and Imitation from Humans

IsaacLab

Unified framework for robot learning built on NVIDIA Isaac Sim

KataGo

GTP engine and self-play learning in Go

legged_gym

Isaac Gym Environments for Legged Robots

lerobot

🤗 LeRobot: State-of-the-art Machine Learning for Real-World Robotics in Pytorch

LightZero

[NeurIPS 2023 Spotlight] LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Sequential Decision Scenarios

ManiSkill

SAPIEN Manipulation Skill Framework, a GPU parallelized robotics simulator and benchmark

metadrive

MetaDrive: Open-source driving simulator

MiniCPM

MiniCPM-2B: An end-side LLM outperforms Llama2-13B.

mujoco

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

openvla

OpenVLA: An open-source vision-language-action model for robotic manipulation.

posteriors

Uncertainty quantification with PyTorch

psopt

PSOPT Optimal Control Software

robocasa

RoboCasa: Large-Scale Simulation of Everyday Tasks for Generalist Robots

robomimic

robomimic: A Modular Framework for Robot Learning from Demonstration

robosuite

robosuite: A Modular Simulation Framework and Benchmark for Robot Learning

robotics_transformer

scenarionet

ScenarioNet: Scalable Traffic Scenario Management System for Autonomous Driving

SimplerEnv

Evaluating and reproducing real-world robot manipulation policies (e.g., RT-1, RT-1-X, Octo) in simulation under common setups (e.g., Google Robot, WidowX+Bridge)

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

UniTraj

A Unified Framework for scalable Vehicle Trajectory Prediction