9c143c915c
Diff: https://github.com/jax-ml/jax/compare/jax-v0.5.1...jax-v0.5.2 Changelog: https://github.com/jax-ml/jax/releases/tag/jax-v0.5.2