diff --git a/AI-and-Analytics/Features-and-Functionality/IntelTensorFlow_Transformer_AMX_bfloat16_MixedPrecision/pyproject.toml b/AI-and-Analytics/Features-and-Functionality/IntelTensorFlow_Transformer_AMX_bfloat16_MixedPrecision/pyproject.toml
index aeb71d26e5..683e46a7ce 100644
--- a/AI-and-Analytics/Features-and-Functionality/IntelTensorFlow_Transformer_AMX_bfloat16_MixedPrecision/pyproject.toml
+++ b/AI-and-Analytics/Features-and-Functionality/IntelTensorFlow_Transformer_AMX_bfloat16_MixedPrecision/pyproject.toml
@@ -10,7 +10,7 @@ readme = "README.md"
 requires-python = ">=3.11,<3.12"
 dependencies = [
     "intel-extension-for-tensorflow[cpu]>=2.15.0",
-    "keras==3.0",
+    "keras==3.11.0",
     "neural-compressor>=3.3",
     "onnxruntime>=1.20.1",
 ]