JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
Why do you think that https://github.com/google-deepmind/optax is a good alternative to jaxrl
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
Why do you think that https://github.com/google-deepmind/optax is a good alternative to jaxrl