diff --git a/requirements.txt b/requirements.txt
index be5b9f9..9788f34 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
 torch>=2.0
 tqdm
-numpy<2
+numpy>=2
 matplotlib
 pyro-ppl
 scipy