diff --git a/pyproject.toml b/pyproject.toml
index de7b51862..c38f9ac5a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -31,7 +31,7 @@ dependencies = [
   "cola-ml==0.0.5",
   "jaxopt==0.8.2",
   "flax>=0.8.5",
-  "numpy<2.0.0",
+  "numpy",
 ]
 
 [tool.hatch.envs.default]