diff --git a/python/requirements/tune/requirements_tune.txt b/python/requirements/tune/requirements_tune.txt
index 30f87bc10..588c00162 100644
--- a/python/requirements/tune/requirements_tune.txt
+++ b/python/requirements/tune/requirements_tune.txt
@@ -1,6 +1,6 @@
 autogluon.core==0.1.0
 ax-platform==0.1.9; python_version < '3.7'
-ax-platform==0.1.20; python_version >= '3.7'
+ax-platform==0.2.1; python_version >= '3.7'
 bayesian-optimization==1.2.0
 ConfigSpace==0.4.18
 dask[complete]==2021.03.0; python_version < '3.7'