diff --git a/pyproject.toml b/pyproject.toml
index 7000bf52..5fe07ab6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -57,7 +57,7 @@ dependencies = [
     "pydantic",
     "torchdata",
     "boto3",
-    "numpy<2.0.0"
+    "numpy<3.0.0"
 ]
 version = "1.1.3"
 
diff --git a/requirements.txt b/requirements.txt
index 1f8d685f..8d7c7d8d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,5 +6,5 @@ ase==3.23.0
 pydantic==2.8.2
 torchdata==0.8.0
 boto3==1.34.159
-numpy==1.26.4
+numpy==2.0.1
 sympy==1.13.2