diff --git a/requirements/TUNE_REQUIREMENTS.txt b/requirements/TUNE_REQUIREMENTS.txt
index d8e6d773..ab4222a3 100644
--- a/requirements/TUNE_REQUIREMENTS.txt
+++ b/requirements/TUNE_REQUIREMENTS.txt
@@ -2,4 +2,4 @@ mypy_extensions==0.4.3; python_version < '3.8'
 optuna==2.9.1
 torchvision==0.9.1
 torch==1.8.1
-ax-platform==0.2.0; python_version >= '3.7'
\ No newline at end of file
+ax-platform==0.2.1; python_version >= '3.7'
\ No newline at end of file