rlax (pypi): A library of reinforcement learning building blocks in JAX.
Keywords: reinforcement-learning|python|machine|learning