diff --git a/setup.py b/setup.py
index 3432b37..cde6dd2 100644
--- a/setup.py
+++ b/setup.py
@@ -9,13 +9,13 @@
     "torchvision>=0.11.0",
     "lightning-bolts",
     "matplotlib",
-    "numpy<2.0.0",  # Due to pandas incompatibility
+    "numpy",
     "scipy",
     "scikit-learn",
     "scprep",
     "scanpy",
     "torchdyn",
-    "pot<0.9.5",
+    "pot",
     "torchdiffeq",
     "absl-py",
     "clean-fid",