JAX v0.5.0

As of this release, JAX now uses effort-based versioning. Since this release makes a breaking change to PRNG key semantics that may require users to update their code, we are bumping the “meso” version of JAX to signify this.

https://github.com/jax-ml/jax/releases/tag/jax-v0.5.0