diff --git a/pixi.lock b/pixi.lock index 612ec955..fc819575 100644 --- a/pixi.lock +++ b/pixi.lock @@ -5232,7 +5232,7 @@ packages: - pypi: . name: array-api-extra version: 0.7.2.dev0 - sha256: dec3f5c0e283642441e3d0d2d24133e081494bb26fc1b8b4e33e1cc5eae06162 + sha256: 3fad55a80a112d7bbf96ef291d9d500be2a20e5b72f3f17159f6cdb9d083e40a requires_dist: - array-api-compat>=1.11.2,<2 requires_python: '>=3.10' diff --git a/pyproject.toml b/pyproject.toml index 7d81fe17..39bc5bea 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -134,7 +134,7 @@ numpy = "=1.22.0" # Backends that can run on CPU-only hosts # Note: JAX and PyTorch will install CPU variants. [tool.pixi.feature.backends.dependencies] -pytorch = ">=1.13.1" +pytorch = ">=2.6.0" dask = ">=2025.3.0" numba = ">=0.61.0" # sparse dependency llvmlite = ">=0.44.0" # sparse dependency @@ -166,7 +166,7 @@ system-requirements = { cuda = "12" } [tool.pixi.feature.cuda-backends.target.linux-64.dependencies] cupy = ">=13.4.1" jaxlib = { version = ">=0.5.2", build = "cuda12*" } -pytorch = { version = ">=1.13.1", build = "cuda12*" } +pytorch = { version = ">=2.6.0", build = "cuda12*" } [tool.pixi.feature.cuda-backends.target.osx-64.dependencies] # cupy = "*" # unavailable @@ -181,7 +181,7 @@ pytorch = { version = ">=1.13.1", build = "cuda12*" } [tool.pixi.feature.cuda-backends.target.win-64.dependencies] cupy = ">=13.4.1" # jaxlib = { version = "*", build = "cuda12*" } # unavailable -pytorch = { version = ">=1.13.1", build = "cuda12*" } +pytorch = { version = ">=2.6.0", build = "cuda12*" } [tool.pixi.environments] default = { features = ["py313"], solve-group = "py313" }