jaxplore • 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 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).

Insights and articles from experts in the field.

4000x Speedups with PureJaxRL

Achieving Over 4000x Speedups and Meta-Evolving Discoveries with PureJaxRL.
Earl: a framework for scalable reinforcement learning research

This blog post describes Earl, a reinforcement learning (RL) framework that enables scalable distributed training across multiple devices.
paperswithcode.com - Reinforcement Learning

Reinforcement Learning section on paperwithcode.com
Spinning Up in Deep RL

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