chore(deps-dev): update jax requirement from <=0.2.12 to <=0.5.0

Updates the requirements on [jax](https://github.com/jax-ml/jax) to permit the latest version.
- [Release notes](https://github.com/jax-ml/jax/releases)
- [Changelog](https://github.com/jax-ml/jax/blob/main/CHANGELOG.md)
- [Commits](jax-ml/jax@jaxlib-v0.1.32...jax-v0.5.0)

---
updated-dependencies:
- dependency-name: jax
  dependency-type: direct:development
...

Signed-off-by: dependabot[bot] <[email protected]>

setup.cfg

@@ -66,7 +66,7 @@ GBDT =
     lightgbm < 4
 neural_tangents =
     jaxlib <= 0.1.65
-    jax <= 0.2.12
+    jax <= 0.5.0
     neural_tangents < 0.6.0
 gpflow =
     tensorflow >= 2.5, < 2.9
@@ -80,7 +80,7 @@ all =
     matplotlib
     lightgbm < 4
     jaxlib <= 0.1.65
-    jax <= 0.2.12
+    jax <= 0.5.0
     neural_tangents < 0.6.0
     tensorflow >= 2.5, < 2.9
     tensorflow-probability >= 0.12, < 0.13