RLax ---- RLax (pronounced "relax") is a library built on top of JAX that exposes …
https://github.com/deepmind/rlax
latest