@@ -134,7 +134,7 @@ numpy = "=1.22.0"
134134# Backends that can run on CPU-only hosts
135135# Note: JAX and PyTorch will install CPU variants.
136136[tool .pixi .feature .backends .dependencies ]
137- pytorch = " >=1.13.1 "
137+ pytorch = " >=2.6.0 "
138138dask = " >=2025.3.0"
139139numba = " >=0.61.0" # sparse dependency
140140llvmlite = " >=0.44.0" # sparse dependency
@@ -166,7 +166,7 @@ system-requirements = { cuda = "12" }
166166[tool .pixi .feature .cuda-backends .target .linux-64 .dependencies ]
167167cupy = " >=13.4.1"
168168jaxlib = { version = " >=0.5.2" , build = " cuda12*" }
169- pytorch = { version = " >=1.13.1 " , build = " cuda12*" }
169+ pytorch = { version = " >=2.6.0 " , build = " cuda12*" }
170170
171171[tool .pixi .feature .cuda-backends .target .osx-64 .dependencies ]
172172# cupy = "*" # unavailable
@@ -181,7 +181,7 @@ pytorch = { version = ">=1.13.1", build = "cuda12*" }
181181[tool .pixi .feature .cuda-backends .target .win-64 .dependencies ]
182182cupy = " >=13.4.1"
183183# jaxlib = { version = "*", build = "cuda12*" } # unavailable
184- pytorch = { version = " >=1.13.1 " , build = " cuda12*" }
184+ pytorch = { version = " >=2.6.0 " , build = " cuda12*" }
185185
186186[tool .pixi .environments ]
187187default = { features = [" py313" ], solve-group = " py313" }
0 commit comments