diff --git a/jaxlib/setup.py b/jaxlib/setup.py index 788f3fd4d..8449f9d5b 100644 --- a/jaxlib/setup.py +++ b/jaxlib/setup.py @@ -60,7 +60,12 @@ setup( author_email='jax-dev@google.com', packages=['jaxlib', 'jaxlib.xla_extension'], python_requires='>=3.9', - install_requires=['scipy>=1.7', 'numpy>=1.22', 'ml_dtypes>=0.2.0'], + install_requires=[ + 'scipy>=1.9', + "scipy>=1.11.1; python_version>='3.12'", + 'numpy>=1.22', + 'ml_dtypes>=0.2.0', + ], extras_require={ 'cuda11_pip': [ "nvidia-cublas-cu11>=11.11", diff --git a/setup.py b/setup.py index 93e7f36e4..7f1b4bd52 100644 --- a/setup.py +++ b/setup.py @@ -78,7 +78,8 @@ setup( "numpy>=1.23.2; python_version>='3.11'", "numpy>=1.26.0; python_version>='3.12'", 'opt_einsum', - 'scipy>=1.7', + 'scipy>=1.9', + "scipy>=1.11.1; python_version>='3.12'", # Required by xla_bridge.discover_pjrt_plugins for forwards compat with # Python versions < 3.10. Can be dropped when 3.10 is the minimum # required Python version.