diff --git a/pyproject.toml b/pyproject.toml
index 73fa130..b6bfff5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -39,7 +39,7 @@ dependencies = [
     "typer",
     "lightning",
     "torch>=2.3.0",  # needed for numpy 2.0
-    "jaxtyping",
+    "jaxtyping<=0.2.34",
     "wandb",
     "tqdm",
     "moviepy>=1.0.3",