diff --git a/paxml/pip_package/requirements.txt b/paxml/pip_package/requirements.txt
index f6c304a4..93fb1fc0 100644
--- a/paxml/pip_package/requirements.txt
+++ b/paxml/pip_package/requirements.txt
@@ -310,7 +310,7 @@ jupyterlab-server==2.27.3
     #   notebook
 jupyterlab-widgets==3.0.15
     # via ipywidgets
-keras==2.9.0
+keras==3.13.2
     # via tensorflow
 keras-preprocessing==1.1.2
     # via tensorflow