JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
Why do you think that https://github.com/google/jax 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/jax is a good alternative to jaxrl