f35542a2c7
Diff: https://github.com/jax-ml/jax/compare/jax-v0.5.2...jax-v0.5.3 Changelog: https://github.com/jax-ml/jax/releases/tag/jax-v0.5.3