From 12f08820d50b1f327f98712ffbc88c4052fff2fd Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Thu, 16 Jul 2026 17:21:31 -0400 Subject: [PATCH] JAX: add v0.11.0 Signed-off-by: Adam J. Stewart --- repos/spack_repo/builtin/packages/py_jax/package.py | 8 +++++++- repos/spack_repo/builtin/packages/py_jaxlib/package.py | 2 ++ 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/repos/spack_repo/builtin/packages/py_jax/package.py b/repos/spack_repo/builtin/packages/py_jax/package.py index f4931b9831a..b09c17a6f75 100644 --- a/repos/spack_repo/builtin/packages/py_jax/package.py +++ b/repos/spack_repo/builtin/packages/py_jax/package.py @@ -23,6 +23,7 @@ class PyJax(PythonPackage): tags = ["e4s"] + version("0.11.0", sha256="a2feb7cfa48bb35d36b8ecec16a4ec24044dec01935f779b28b017213697b195") version("0.10.2", sha256="bf77428a8c2e6904c4f46d5ab12aa5cfc6cad2179f07f7e4c0fc75ac86ef0639") version("0.10.1", sha256="11672410faf8752429eb9a131de203dc488a2a3a012d509baa2b39878008810d") version("0.10.0", sha256="0119c767de1645f407df72345d28a3837dc904f1d698911c121d8f2b396fdece") @@ -84,6 +85,7 @@ class PyJax(PythonPackage): with default_args(type=("build", "run")): # setup.py + depends_on("python@3.12:", when="@0.11:") depends_on("python@3.11:", when="@0.7:") depends_on("python@3.10:", when="@0.4.31:") depends_on("python@3.9:", when="@0.4.14:") @@ -92,7 +94,8 @@ class PyJax(PythonPackage): depends_on("py-ml-dtypes@0.2:", when="@0.4.14:") depends_on("py-ml-dtypes@0.1:", when="@0.4.9:") depends_on("py-ml-dtypes@0.0.3:", when="@0.4.7:") - depends_on("py-numpy@2:", when="@0.7.2:") + depends_on("py-numpy@2.1:", when="@0.11:") + depends_on("py-numpy@2.0:", when="@0.7.2:") depends_on("py-numpy@1.26:", when="@0.6.2:") depends_on("py-numpy@1.25:", when="@0.5:") depends_on("py-numpy@1.24:", when="@0.4.31:") @@ -102,6 +105,7 @@ class PyJax(PythonPackage): # https://github.com/google/jax/issues/19246 depends_on("py-numpy@:1", when="@:0.4.25") depends_on("py-opt-einsum") + depends_on("py-scipy@1.15:", when="@0.11:") depends_on("py-scipy@1.14:", when="@0.10:") depends_on("py-scipy@1.13:", when="@0.7.2:") depends_on("py-scipy@1.12:", when="@0.6.2:") @@ -114,6 +118,7 @@ class PyJax(PythonPackage): # jax/_src/lib/__init__.py # https://github.com/google/jax/commit/8be057de1f50756fe7522f7e98b2f30fad56f7e4 for v in [ + "0.11.0", "0.10.2", "0.10.1", "0.10.0", @@ -174,6 +179,7 @@ class PyJax(PythonPackage): depends_on(f"py-jaxlib@:{v}", when=f"@{v}") # See _minimum_jaxlib_version in jax/version.py + depends_on("py-jaxlib@0.11.0:", when="@0.11.0:") depends_on("py-jaxlib@0.10.2:", when="@0.10.2:") depends_on("py-jaxlib@0.10.1:", when="@0.10.1:") depends_on("py-jaxlib@0.10.0:", when="@0.10.0:") diff --git a/repos/spack_repo/builtin/packages/py_jaxlib/package.py b/repos/spack_repo/builtin/packages/py_jaxlib/package.py index 221785c338e..c5ae25dd3a1 100644 --- a/repos/spack_repo/builtin/packages/py_jaxlib/package.py +++ b/repos/spack_repo/builtin/packages/py_jaxlib/package.py @@ -49,6 +49,7 @@ class PyJaxlib(PythonPackage, CudaPackage, ROCmPackage): license("Apache-2.0") maintainers("adamjstewart", "jonas-eschle") + version("0.11.0", sha256="007ef373573ff2fb8a5485679b791581fda328754fd7ae491de3bcdb0fc70d07") version("0.10.2", sha256="fa7214ab31ed1cd418b4305807e9c4f3f175c783eeea40c28e0f77c3f4c24bc7") version("0.10.1", sha256="15983d01b0c858738b16b19b773459d22449992ce1ee97688cc532ea0047de9e") version("0.10.0", sha256="12ae17617d1346e2f98cfc48c1a000adc7389784eb119e8108a22dfd57cbb8c3") @@ -152,6 +153,7 @@ class PyJaxlib(PythonPackage, CudaPackage, ROCmPackage): with default_args(type=("build", "run")): # Based on PyPI wheels + depends_on("python@3.12:", when="@0.11:") depends_on("python@3.11:", when="@0.7:") depends_on("python@3.10:", when="@0.4.31:") depends_on("python@3.9:", when="@0.4.14:")