diff --git a/setup.py b/setup.py
index 3907d09..362f8db 100644
--- a/setup.py
+++ b/setup.py
@@ -41,8 +41,8 @@ def getVersionNumber():
     install_requires=[
         "reciprocalspaceship>=0.9.16",
         "tqdm",
-        "tensorflow==2.12",
-        "tensorflow-probability==0.20.0",
+        "tensorflow>=2.8",
+        "tensorflow-probability",
         "matplotlib",
         "seaborn",
     ],