diff --git a/requirements/TUNE_REQUIREMENTS.txt b/requirements/TUNE_REQUIREMENTS.txt
index f9913563..954ef1ce 100644
--- a/requirements/TUNE_REQUIREMENTS.txt
+++ b/requirements/TUNE_REQUIREMENTS.txt
@@ -1,4 +1,4 @@
 mypy_extensions~=0.4; python_version < '3.8'
 optuna~=2.10
 torch>=1.8.1
-ax-platform~=0.2.3; python_version >= '3.7'
+ax-platform~=0.2.4; python_version >= '3.7'