This projects contains implementations of various reinforcement learning algorithms ported to JAX and Equinox.
https://github.com/dorjeduck/eqxrl