🦁 A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX

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

jax-triton contains integrations between JAX and OpenAI Triton

🐈‍⬛ Contextual bandits library for continuous action trees with smoothing in JAX

Accelerated Quality-Diversity

OpenRAVE Controller Plugin for UR5 (Universal Robots UR5) Robot