Lists (9)
Starred repositories
🤗 LeRobot: Making AI for Robotics more accessible with end-to-end learning
Python Fire is a library for automatically generating command line interfaces (CLIs) from absolutely any Python object.
Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.
Codebase for Berkeley Humanoid Lite
Probabilistic reasoning and statistical analysis in TensorFlow
Flax is a neural network library for JAX that is designed for flexibility.
A JAX research toolkit for building, editing, and visualizing neural networks.
Unified Implementations of Offline Reinforcement Learning Algorithms
Scenic: A Jax Library for Computer Vision Research and Beyond
A tool for aggregating and plotting MARL experiment data.
Graph Neural Network Library for PyTorch
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
Official implementation of Diffusion Policy Policy Optimization, arxiv 2024
Code for the paper "Planning with Diffusion for Flexible Behavior Synthesis"
[RSS 2023] Diffusion Policy Visuomotor Policy Learning via Action Diffusion
High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)
Model-based design and verification for robotics.
An energy-based formulation for soft-bodied virtual creatures
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
A collection of MARL benchmarks based on TorchRL
Adjoint-Equivariant Neural Network for Semi-Simple Lie Algebras