Merge branch 'main' into batched-inference-and-padding

Author: MMathisLab

.github/workflows/docs.yml

@@ -47,11 +47,7 @@ jobs:
         with:
           repository: AdaptiveMotorControlLab/cebra-demos
           path: docs/source/demo_notebooks
-          # NOTE(stes): This is a temporary branch to add the xCEBRA demo notebooks
-          # to the docs. Once the notebooks are merged into main, we can remove this
-          # branch and change the ref to main.
-          # ref: main
-          ref: stes/add-xcebra
+          ref: main
 
       - name: Set up Python 3.10
         uses: actions/setup-python@v5

docs/source/conf.py

@@ -269,6 +269,8 @@ def get_years(start_year=2021):
         "_static/thumbnails/openScope_demo.png",
     "demo_notebooks/Demo_dandi_NeuroDataReHack_2023":
         "_static/thumbnails/dandi_demo_monkey.png",
+    "demo_notebooks/Demo_xCEBRA_RatInABox":
+        "_static/thumbnails/xCEBRA.png"
 }
 
 rst_prolog = r"""

docs/source/installation.rst

@@ -149,6 +149,13 @@ we outline different options below.
 
         Note that, similarly to that last command, you can select the specific install options of interest based on their description above and on your usecase.
 
+    .. tab:: Docker
+
+        .. code:: bash
+
+            $ docker pull mmathislab/cebra-cuda12.4-cudnn9
+
+        You can pull our container from DockerHub: https://hub.docker.com/u/mmathislab
 ..