btx0424's repositories
tensordict
TensorDict is a pytorch dedicated tensor container.
Orbit
Unified framework for robot learning built on NVIDIA Isaac Sim
dial-mpc
Official implementation for the paper "Full-Order Sampling-Based MPC for Torque-Level Locomotion Control via Diffusion-Style Annealing". DIAL-MPC is a novel sampling-based MPC framework for legged robot full-order torque-level control with both precision and agility in a training-free manner.
rl
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
brax
Massively parallel rigidbody physics simulation on accelerator hardware.
diffusion_policy
[RSS 2023] Diffusion Policy Visuomotor Policy Learning via Action Diffusion
dreamerv3-torch
Implementation of Dreamer v3 in pytorch.
ManiSkill
SAPIEN Manipulation Skill Framework, a GPU parallelized robotics simulator and benchmark
unitree_sdk2
Unitree robot sdk version 2. https://support.unitree.com/home/zh/developer
diffusion-forcing
code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"
tdmpc2
Code for "TD-MPC2: Scalable, Robust World Models for Continuous Control"
gymnax
RL Environments in JAX 🌍
IsaacGymEnvs
Isaac Gym Reinforcement Learning Environments
RODE
Codes accompanying the paper "RODE: Learning Roles to Decompose Multi-Agent Tasks (ICLR 2021, https://arxiv.org/abs/2010.01523). RODE is a scalable role-based multi-agent learning method which effectively discovers roles based on joint action space decomposition according to action effects, establishing a new state of the art on the StarCraft multi-agent benchmark.
gym-pybullet-drones
PyBullet Gym environments for single and multi-agent reinforcement learning of quadcopter control
dreamerv2-pbt
Mastering Atari with Discrete World Models
on-policy
This is the official implementation of Multi-Agent PPO (MAPPO).