Merge branch 'add_py-jaxlib_cuda_variant_and_disable' into 'master'

py-jaxlib: Add explicit variant `cuda` (but disable)

See merge request technical-coordination/project-internal/devops/platform/ebrains-spack-builds!278

packages.yaml

 packages:
     all:
+        # collab-specific settings
         target: [x86_64]
         compiler: [gcc@10.3.0]
     python:
+        # collab-specific settings
         version: [3.8.11]
     r:
         version: [4.1.3]
+        # EM: +X fixes build for collab
         variants: [+X]
     py-torch:
+        # ECM: keep at latest w.r.t. tracked upstream spack
         version: [1.11.0]
+        # ECM: No cuda nor rocm in collab VMs.
+        #      I did not try to minimize the specification any further.
         variants: [~cuda~rocm~valgrind~mkldnn~mpi~gloo+tensorpipe~onnx_ml]
     py-jax:
+        # ECM: No cuda in collab VMs.
+        variants: [~cuda]
+    py-jaxlib:
+        # ECM: No cuda in collab VMs.
         variants: [~cuda]

packages/py-jaxlib/package.py

@@ -21,6 +21,8 @@ class PyJaxlib(PythonPackage, CudaPackage):
     version("0.3.25", sha256="73ebc7868631cd9d520385557bbd7f08762d748a5a6a1bebef0f3b8d7ba748ef")
     version("0.1.74", sha256="bbc78c7a4927012dcb1b7cd135c7521f782d7dad516a2401b56d3190f81afe35")
 
+    variant("cuda", default=True, description="Build with CUDA")
+
     depends_on("python@3.7:", type=("build", "run"))
     depends_on("py-setuptools", type="build")
     depends_on("py-numpy@1.18:", type=("build", "run"))