DEV: remove redundant CUDA pins for JAX

Author: crusaderky

pixi.lock

@@ -132,6 +132,7 @@ python = "~=3.13.0"
 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 = "*"
 dask = "*"
@@ -154,24 +155,22 @@ jax = "*"
 # jax = "*"  # unavailable
 
 # Backends that require a GPU host and a CUDA driver
+# Note: JAX and PyTorch automatically install CUDA variants
+# thanks to the `system-requirements` below.
 [tool.pixi.feature.cuda-backends]
 system-requirements = { cuda = "12" }
 
 [tool.pixi.feature.cuda-backends.target.linux-64.dependencies]
 cupy = "*"
-jaxlib = { version = "*", build = "cuda12*" }
 
 [tool.pixi.feature.cuda-backends.target.osx-64.dependencies]
 # cupy = "*"  # unavailable
-# jaxlib = { version = "*", build = "cuda12*" }  # unavailable
 
 [tool.pixi.feature.cuda-backends.target.osx-arm64.dependencies]
 # cupy = "*"  # unavailable
-# jaxlib = { version = "*", build = "cuda12*" }  # unavailable
 
 [tool.pixi.feature.cuda-backends.target.win-64.dependencies]
 cupy = "*"
-# jaxlib = { version = "*", build = "cuda12*" }  # unavailable
 
 [tool.pixi.environments]
 default = { features = ["py313"], solve-group = "py313" }