Jumanji

github.comtool

A Suite of Industry-Driven Hardware-Accelerated RL Environments written in JAX.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   JAX RL github.com

   tool category #jax link ↗

   Implementations of reinforcement learning algorithms.

   
2. ▲
   purejaxrl github.com

   tool category #jax link ↗

   Vectorisable, end-to-end RL algorithms in JAX.

   
3. ▲
   RLax github.com

   tool category #jax link ↗

   Library for implementing reinforcement learning agents.

4. ▲
   Introduction to JAX by Kevin Murphy colab.research.google.com

   site category #jax link ↗

   Colab that introduces various aspects of the language and applies them to simple ML problems.

   
5. ▲
   Pgx github.com

   tool category #jax link ↗

   Vectorized board game environments for RL with an AlphaZero example.

   
6. ▲
   JAX Toolbox github.com

   tool category #jax link ↗

   Nightly CI and optimized examples for JAX on NVIDIA GPUs using libraries such as T5x, Paxml, and Transformer Engine.

   

JAX › Libraries > New Libraries: “A Suite of Industry-Driven Hardware-Accelerated RL Environments written in JAX.”