diff --git a/setup.cfg b/setup.cfg index 321020ad9..eb570dafb 100644 --- a/setup.cfg +++ b/setup.cfg @@ -121,6 +121,8 @@ jax_core_deps = chex==0.1.7 ml_dtypes==0.2.0 protobuf==4.25.3 + scipy==1.11.4 + # JAX CPU jax_cpu =