dejax (pypi): Accelerated replay buffers in JAX.
Keywords: jax|reinforcement-learning