diff --git a/setup.py b/setup.py
index eb87e43..1128164 100644
--- a/setup.py
+++ b/setup.py
@@ -32,9 +32,12 @@
     install_requires=[
         "toolz",
         "colour",
-        "svgwrite",
-        "Pillow",
-        "loguru",
+        "imageio",
+        "numpy",
+        "matplotlib",
+        "jax",
+        "chex",
+        "jaxtyping",
         "typing-extensions",
         "importlib-metadata",
     ],