Really Fast End-to-End Jax RL Implementations

add statistical significance annotations on seaborn plots. Further development of statannot, with bugfixes, new features, and a different API.

🕹️ A diverse suite of scalable reinforcement learning environments in JAX

Code for "TD-MPC2: Scalable, Robust World Models for Continuous Control"

An extension of the PyMARL codebase that includes additional algorithms and environment support

Official code repo for the MARL book (www.marl-book.com)

Implementation of paper "Towards a Unified View of Parameter-Efficient Transfer Learning" (ICLR 2022)

Implementation of π₀, the robotic foundation model architecture proposed by Physical Intelligence

🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL

Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!

Pyrallis is a framework for structured configuration parsing from both cmd and files. Simply define your desired configuration structure as a dataclass and let pyrallis do the rest!

Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.

Simple single-file baselines for Q-Learning in pure-GPU setting

Realtime API for Lucky World simulator with ROS-like interface

A Python package of computer vision models for the Equinox ecosystem.

JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"

[ICCV 25] Official repository of "Collaborative Instance Object Navigation: Leveraging Uncertainty-Awareness to Minimize Human-Agent Dialogues"

Simple package implementing defeasible logic