diff --git a/pyproject.toml b/pyproject.toml
index 0ac2a0c7..09f77296 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -34,6 +34,7 @@ dependencies = [
   "diffrax",
   "equinox",
   "jax",
+  "jaxlib",
   "typing_extensions",
 ]