diff --git a/pyproject.toml b/pyproject.toml index 69f1fca..bd4d91f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -177,7 +177,7 @@ system-requirements = { cuda = "12" } [tool.pixi.feature.cuda-backends.target.linux-64.dependencies] cupy = ">=13.5.1" -jaxlib = { version = ">=0.6.0", build = "cuda12*" } +jaxlib = { version = ">=0.7.0", build = "cuda12*" } pytorch = { version = ">=2.7.1", build = "cuda12*" } [tool.pixi.feature.cuda-backends.target.osx-64.dependencies]