diff --git a/paxml/pip_package/requirements.txt b/paxml/pip_package/requirements.txt
index 0a1675e5..df9625bb 100644
--- a/paxml/pip_package/requirements.txt
+++ b/paxml/pip_package/requirements.txt
@@ -658,7 +658,7 @@ tensorboard-data-server==0.6.1
     # via tensorboard
 tensorboard-plugin-wit==1.8.1
     # via tensorboard
-tensorflow==2.9.3
+tensorflow==2.12.1
     # via
     #   -r paxml-requirements.in
     #   -r praxis-requirements.in