diff --git a/pyproject.toml b/pyproject.toml
index 49d828a57a7094521f926ae6e9679ab1c5d66db7..2223f3d6752145782324a1bae640a0e00401afa9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -53,7 +53,7 @@ all = [  # all non-tests extra dependencies
     "jax[cpu] ~= 0.4.1",
     "opacus ~= 1.4",
     "protobuf >= 3.19",
-    "tensorflow ~= 2.5",
+    "tensorflow ~= 2.11, < 2.16",
     "torch >= 1.13, < 3.0",
     "websockets >= 10.1, < 13.0",
 ]
@@ -71,7 +71,7 @@ haiku = [
     "jax[cpu] ~= 0.4.1",  # NOTE: GPU support must be manually installed
 ]
 tensorflow = [
-    "tensorflow ~= 2.5",
+    "tensorflow ~= 2.11, < 2.16",  # starting with 2.16 TF upgrades to Keras 3
 ]
 torch = [  # generic requirements for Torch
     "torch >= 1.13, < 3.0",