diff --git a/.readthedocs.yml b/.readthedocs.yml
index 85655289..b2597e97 100644
--- a/.readthedocs.yml
+++ b/.readthedocs.yml
@@ -13,3 +13,4 @@ python:
   install:
     - requirements: requirements/runtime.txt
     - requirements: requirements/docs.txt
+    - requirements: requirements/docs_extra.txt
diff --git a/requirements/docs.txt b/requirements/docs.txt
index 1507e3d7..a0d0e050 100644
--- a/requirements/docs.txt
+++ b/requirements/docs.txt
@@ -1,4 +1,3 @@
-deepspeed
 docutils==0.17.1
 myst-parser
 opencv-python
diff --git a/requirements/docs_extra.txt b/requirements/docs_extra.txt
new file mode 100644
index 00000000..0f2c819d
--- /dev/null
+++ b/requirements/docs_extra.txt
@@ -0,0 +1 @@
+deepspeed
diff --git a/requirements/tests.txt b/requirements/tests.txt
index 599163fd..544ddba9 100644
--- a/requirements/tests.txt
+++ b/requirements/tests.txt
@@ -1,4 +1,4 @@
-aim;sys_platform!='win32'
+aim<=3.17.5;sys_platform!='win32'
 bitsandbytes
 clearml
 coverage