build(deps-dev): bump jax from 0.5.0 to 0.6.2

Bumps [jax](https://github.com/jax-ml/jax) from 0.5.0 to 0.6.2.
- [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.5.0...jax-v0.6.2)

---
updated-dependencies:
- dependency-name: jax
  dependency-version: 0.6.2
  dependency-type: direct:development
  update-type: version-update:semver-minor
...

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

Author: dependabot[bot]

pyproject.toml

@@ -169,7 +169,7 @@ pin_pytorch_gpu = [
   "torch>=2.7,<2.10",
 ]
 pin_jax = [
-  "jax==0.5.0;python_version>='3.10'",
+  "jax==0.6.2;python_version>='3.10'",
 ]
 
 [tool.setuptools_scm]