diff --git a/tools/ray_tpu/src/tune/requirements.txt b/tools/ray_tpu/src/tune/requirements.txt
index 12b80af8..87c77312 100644
--- a/tools/ray_tpu/src/tune/requirements.txt
+++ b/tools/ray_tpu/src/tune/requirements.txt
@@ -8,7 +8,7 @@ ml-collections==0.1.0
 numpy==1.22.0
 optax==0.1.0
 pandas
-ray[tune]==2.5.1
+ray[tune]==2.54.0
 tensorflow-cpu==2.9.3
 tensorflow-datasets==4.4.0
 protobuf==3.19.0
\ No newline at end of file