diff --git a/src/sample_pytorch_gpu_project/.devcontainer/requirements.txt b/src/sample_pytorch_gpu_project/.devcontainer/requirements.txt
index ca836da..3ac9348 100644
--- a/src/sample_pytorch_gpu_project/.devcontainer/requirements.txt
+++ b/src/sample_pytorch_gpu_project/.devcontainer/requirements.txt
@@ -1,5 +1,5 @@
 torch==2.11.0
 torchvision==0.26.0
 # the below are used by AML, can be deleted if not using AML
-mlflow==3.9.0
+mlflow==3.11.0rc1
 azureml-mlflow==1.62.0.post2