diff --git a/.github/workflows/mkdocs-main.yml b/.github/workflows/mkdocs-main.yml
index a77180fc5..6e0cb2b6e 100644
--- a/.github/workflows/mkdocs-main.yml
+++ b/.github/workflows/mkdocs-main.yml
@@ -7,7 +7,7 @@ jobs:
     runs-on: ubuntu-latest
 
     steps:
-      - uses: actions/checkout@v2
+      - uses: actions/checkout@v4
         with:
           fetch-depth: 0
 
@@ -15,9 +15,9 @@ jobs:
         working-directory: ./java
         run: echo "DEV_VERSION=$(mvn org.apache.maven.plugins:maven-help-plugin:2.1.1:evaluate -Dexpression=project.version | grep -Ev 'Download|INFO|WARNING')" >> $GITHUB_ENV
 
-      - uses: actions/setup-python@v2
+      - uses: actions/setup-python@v5
         with:
-          python-version: '3.8'
+          python-version: "3.10"
 
       - name: install deps
         working-directory: ./python
@@ -33,4 +33,3 @@ jobs:
 
       - name: mike deploy docs
         run: mike deploy ${{ env.DEV_VERSION }} dev -u
-
diff --git a/.github/workflows/mkdocs-release.yml b/.github/workflows/mkdocs-release.yml
index 44c9e9080..8c4728d02 100644
--- a/.github/workflows/mkdocs-release.yml
+++ b/.github/workflows/mkdocs-release.yml
@@ -2,14 +2,14 @@ name: mkdocs-release
 
 on:
   push:
-    branches: [ branch-* ]
+    branches: [branch-*]
 
 jobs:
   publish-release:
     runs-on: ubuntu-latest
 
     steps:
-      - uses: actions/checkout@v2
+      - uses: actions/checkout@v4
         with:
           fetch-depth: 0
 
@@ -20,16 +20,13 @@ jobs:
       - name: set major/minor release version
         run: echo "MAJOR_VERSION=$(echo $RELEASE_VERSION | sed 's/^\([0-9]*\.[0-9]*\).*$/\1/')" >> $GITHUB_ENV
 
-      - uses: actions/setup-python@v2
+      - uses: actions/setup-python@v5
         with:
-          python-version: '3.8'
+          python-version: "3.10"
 
       - name: install deps
         working-directory: ./python
-        run: cp ../README.md . && pip3 install pip==22.0.3 && pip3 install -e .[dev,docs]
-        
-      - name: use dev mike
-        run: pip3 uninstall -y mike && pip3 install git+'https://github.com/jimporter/mike.git'
+        run: cp ../README.md . && pip3 install -e .[dev,docs]
 
       - name: generate autodoc
         run: python3 auto_doc.py
@@ -43,4 +40,3 @@ jobs:
         run: |
           mike deploy ${{ env.RELEASE_VERSION }} ${{ env.MAJOR_VERSION }} -u --push
           mike alias ${{ env.RELEASE_VERSION }} latest -u --push
-
diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md
index 7e43c1936..58f07bfae 100644
--- a/CONTRIBUTING.md
+++ b/CONTRIBUTING.md
@@ -74,7 +74,7 @@ We use `mkdocs` together with `mike` ([for versioning](https://github.com/jimpor
 1. Currently we are using our own version of `keras-autodoc`
 
     ```bash
-    pip install git+https://github.com/moritzmeister/keras-autodoc@split-tags-properties
+    pip install git+https://github.com/logicalclocks/keras-autodoc
     ```
 
 2. Install HOPSWORKS with `docs` extras:
diff --git a/docs/CONTRIBUTING.md b/docs/CONTRIBUTING.md
index 7e43c1936..58f07bfae 100644
--- a/docs/CONTRIBUTING.md
+++ b/docs/CONTRIBUTING.md
@@ -74,7 +74,7 @@ We use `mkdocs` together with `mike` ([for versioning](https://github.com/jimpor
 1. Currently we are using our own version of `keras-autodoc`
 
     ```bash
-    pip install git+https://github.com/moritzmeister/keras-autodoc@split-tags-properties
+    pip install git+https://github.com/logicalclocks/keras-autodoc
     ```
 
 2. Install HOPSWORKS with `docs` extras:
diff --git a/python/setup.py b/python/setup.py
index b609d9bf0..7ee17747f 100644
--- a/python/setup.py
+++ b/python/setup.py
@@ -31,14 +31,14 @@ def read(fname):
     extras_require={
         "dev": ["pytest", "flake8", "black"],
         "docs": [
-            "mkdocs==1.3.0",
-            "mkdocs-material==8.2.8",
-            "mike==1.1.2",
-            "sphinx==3.5.4",
-            "keras_autodoc @ git+https://git@github.com/moritzmeister/keras-autodoc@split-tags-properties",
-            "markdown-include",
-            "markdown==3.3.7",
-            "pymdown-extensions",
+            "mkdocs==1.5.3",
+            "mkdocs-material==9.5.17",
+            "mike==2.0.0",
+            "sphinx==7.2.6",
+            "keras_autodoc @ git+https://git@github.com/logicalclocks/keras-autodoc",
+            "markdown-include==0.8.1",
+            "markdown==3.6",
+            "pymdown-extensions==10.7.1",
         ],
     },
     author="Logical Clocks AB",