diff --git a/requirements.txt b/requirements.txt
index 37e0801..a8ef762 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,7 +6,7 @@ ml_collections==0.1.1
 absl-py>=0.13.0
 protobuf>=3.20.3
 flatbuffers>=23.1.21
-keras<2.15,>=2.14.0
+keras>=2.14.0,<3.10
 tensorflow-estimator<2.15,>=2.14.0
 zipp>=3.1.0
 httplib2>=0.15.0