Web

Beyond Github and YouTube ..

Official Docs

Links to official documentations for JAX, RL and related tools.

Equinox

Equinox is your one-stop JAX library, for everything you need that isn't already in core JAX.
Flax

Neural Networks for JAX
Jax

Official JAX docs
Optax

Optax is a gradient processing and optimization library for JAX.

Tutorials

Tutorials for learning JAX and Reinforcement Learning.

Introduction to JAX by Google

A beginner-friendly introduction to JAX from the official team.
Spinning Up in Deep RL

Spinning Up is an educational resource produced by OpenAI that makes it easier to learn about deep reinforcement learning (deep RL).

Others

Insights and articles from experts in the field.

4000x Speedups with PureJaxRL

Achieving Over 4000x Speedups and Meta-Evolving Discoveries with PureJaxRL.