diff --git a/e2e/e2e-jax/pyproject.toml b/e2e/e2e-jax/pyproject.toml
index 65d55a243884..b259f66a7bc3 100644
--- a/e2e/e2e-jax/pyproject.toml
+++ b/e2e/e2e-jax/pyproject.toml
@@ -12,7 +12,7 @@ dependencies = [
     "jax==0.4.13",
     "jaxlib==0.4.13",
     "scikit-learn>=1.1.1,<2.0.0",
-    "numpy>=2.0.0",
+    "numpy>=1.21.4,<2.0.0",
 ]
 
 [tool.hatch.build.targets.wheel]