From f03491dd76039bf3638fb92accea0342559d7e8d Mon Sep 17 00:00:00 2001 From: "dependabot[bot]" <49699333+dependabot[bot]@users.noreply.github.com> Date: Sat, 9 May 2026 16:41:15 +0000 Subject: [PATCH] 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](https://github.com/jax-ml/jax/compare/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] --- pyproject.toml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 9262945276..b4c5e523fd 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -176,10 +176,10 @@ pin_pytorch_gpu = [ "torch==2.10.0", ] pin_jax_cpu = [ - "jax==0.5.0;python_version>='3.10'", + "jax==0.6.2;python_version>='3.10'", ] pin_jax_gpu = [ - "jax[cuda12]==0.5.0;python_version>='3.10'", + "jax[cuda12]==0.6.2;python_version>='3.10'", ] [tool.setuptools_scm]