diff --git a/pyproject.toml b/pyproject.toml
index 0d0843e..b5d0bdb 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -14,7 +14,7 @@ requires-python = ">= 3.9,<3.12"
 dependencies = [
     "reciprocalspaceship>=0.9.16",
     "tqdm",
-    "tensorflow",
+    "tensorflow<2.18",
     "tf_keras",
     "tensorflow-probability[tf]",
     "matplotlib",