Fix RTD build

PiperOrigin-RevId: 628863562

README.md

@@ -1,7 +1,7 @@
-<img align="left" width="200" height="200" src="https://raw.githubusercontent.com/google-deepmind/dm_pix/master/dm_pix/images/pix_logo.png" alt="pix_logo"></img>
-
 # PIX
 
+![PIX logo](https://raw.githubusercontent.com/google-deepmind/dm_pix/master/dm_pix/images/pix_logo.png){style="width:200px; height:200px"}
+
 PIX is an image processing library in [JAX], for [JAX].
 
 [![GitHub Workflow CI](https://img.shields.io/github/actions/workflow/status/deepmind/dm_pix/ci.yml?label=pytest&logo=python&logoColor=white&style=flat-square)](https://github.com/deepmind/dm_pix/actions/workflows/ci.yml)

pyproject.toml

@@ -37,29 +37,26 @@ extras = [
     "jax>=0.2.17",
     "jaxlib>=0.1.69",
 ]
-
 test = [
     "scipy",
     "tensorflow",
     "pytest-xdist",
 ]
-
 docs = [
-    "sphinx==4.5.0",
-    "sphinx_rtd_theme==1.0.0",
-    "sphinxcontrib-katex==0.9.0",
-    "sphinxcontrib-bibtex==2.4.2",
-    "sphinx-autodoc-typehints==1.11.1",
-    "IPython==8.10.0",
-    "ipykernel==5.3.4",
-    "pandoc==1.1.0",
-    "myst_nb==0.13.1",
-    "docutils==0.16",
-    "matplotlib==3.5.0",
+    "sphinx>=6.0.0",
+    "sphinx_rtd_theme",
+    "sphinxcontrib-katex",
+    "sphinxcontrib-bibtex",
+    "sphinx-autodoc-typehints",
+    "IPython",
+    "ipykernel",
+    "pandoc",
+    "myst_nb",
+    "docutils",
+    "matplotlib",
 ]
-
 examples = [
-    "Pillow==10.3.0",
+    "Pillow",
 ]
 
 [tool.setuptools.packages.find]