diff --git a/requirements.txt b/requirements.txt
index 96e48e1..ed7169e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -21,9 +21,9 @@ pyspark>=3.2.1
 pyarrow>=1.0.0
 dill==0.3.4
 tqdm==4.62.3
-tensorflow==2.9.3
+tensorflow==2.11.1
 paddlepaddle==2.4.0
 transformers==4.21.1
 jax>=0.3.24
 optax>=0.1.3
-flax>=0.6.2
\ No newline at end of file
+flax>=0.6.2