diff --git a/setup.py b/setup.py
index fb7d133f111..fdbc37d3f2d 100644
--- a/setup.py
+++ b/setup.py
@@ -36,7 +36,7 @@
     "timm",
 ]
 extras["testing"] = extras["test_prod"] + extras["test_dev"]
-extras["deepspeed"] = ["deepspeed<=0.14.0"]
+extras["deepspeed"] = ["deepspeed"]
 extras["rich"] = ["rich"]
 
 extras["test_trackers"] = ["wandb", "comet-ml", "tensorboard", "dvclive"]