Skip to content

Commit f9a6ca4

Browse files
deps: Update jax to >=0.7.0,!=0.6.2
1 parent d59cc8a commit f9a6ca4

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

pyproject.toml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -155,13 +155,13 @@ dask-core = ">=2025.7.0" # No distributed, tornado, etc.
155155
sparse = ">=0.17.0"
156156

157157
[tool.pixi.feature.backends.target.linux-64.dependencies]
158-
jax = ">=0.6.0,!=0.6.2" # 0.6.2 segfaults on Linux CUDA
158+
jax = ">=0.7.0,!=0.6.2" # 0.6.2 segfaults on Linux CUDA
159159

160160
[tool.pixi.feature.backends.target.osx-64.dependencies]
161-
jax = ">=0.6.0,!=0.6.2"
161+
jax = ">=0.7.0,!=0.6.2"
162162

163163
[tool.pixi.feature.backends.target.osx-arm64.dependencies]
164-
jax = ">=0.6.0,!=0.6.2"
164+
jax = ">=0.7.0,!=0.6.2"
165165

166166
[tool.pixi.feature.backends.target.win-64.dependencies]
167167
# jax = "*" # unavailable
@@ -177,7 +177,7 @@ system-requirements = { cuda = "12" }
177177

178178
[tool.pixi.feature.cuda-backends.target.linux-64.dependencies]
179179
cupy = ">=13.5.1"
180-
jaxlib = { version = ">=0.6.0", build = "cuda12*" }
180+
jaxlib = { version = ">=0.7.0", build = "cuda12*" }
181181
pytorch = { version = ">=2.7.1", build = "cuda12*" }
182182

183183
[tool.pixi.feature.cuda-backends.target.osx-64.dependencies]

0 commit comments

Comments
 (0)