Bump jax from 0.4.34 to 0.4.35

Bumps [jax](https://github.com/jax-ml/jax) from 0.4.34 to 0.4.35.
- [Release notes](https://github.com/jax-ml/jax/releases)
- [Changelog](https://github.com/jax-ml/jax/blob/main/CHANGELOG.md)
- [Commits](jax-ml/jax@jax-v0.4.34...jax-v0.4.35)

---
updated-dependencies:
- dependency-name: jax
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>

pyproject.toml

@@ -15,7 +15,7 @@ keywords = []
 
 # pip dependencies of the project
 dependencies = [
-    "jax==0.4.34",
+    "jax==0.4.35",
     "flax==0.10.0",
     "ml_dtypes==0.4.0",
     "optax==0.2.3",