@@ -132,6 +132,7 @@ python = "~=3.13.0"
132
132
numpy = " =1.22.0"
133
133
134
134
# Backends that can run on CPU-only hosts
135
+ # Note: JAX and PyTorch will install CPU variants.
135
136
[tool .pixi .feature .backends .dependencies ]
136
137
pytorch = " *"
137
138
dask = " *"
@@ -154,24 +155,22 @@ jax = "*"
154
155
# jax = "*" # unavailable
155
156
156
157
# Backends that require a GPU host and a CUDA driver
158
+ # Note: JAX and PyTorch automatically install CUDA variants
159
+ # thanks to the `system-requirements` below.
157
160
[tool .pixi .feature .cuda-backends ]
158
161
system-requirements = { cuda = " 12" }
159
162
160
163
[tool .pixi .feature .cuda-backends .target .linux-64 .dependencies ]
161
164
cupy = " *"
162
- jaxlib = { version = " *" , build = " cuda12*" }
163
165
164
166
[tool .pixi .feature .cuda-backends .target .osx-64 .dependencies ]
165
167
# cupy = "*" # unavailable
166
- # jaxlib = { version = "*", build = "cuda12*" } # unavailable
167
168
168
169
[tool .pixi .feature .cuda-backends .target .osx-arm64 .dependencies ]
169
170
# cupy = "*" # unavailable
170
- # jaxlib = { version = "*", build = "cuda12*" } # unavailable
171
171
172
172
[tool .pixi .feature .cuda-backends .target .win-64 .dependencies ]
173
173
cupy = " *"
174
- # jaxlib = { version = "*", build = "cuda12*" } # unavailable
175
174
176
175
[tool .pixi .environments ]
177
176
default = { features = [" py313" ], solve-group = " py313" }
0 commit comments