diff --git a/pyproject.toml b/pyproject.toml
index 152c690b..a73ff6ff 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -36,6 +36,7 @@ dependencies = [
   "coverage",
   "dask",
   "diffrax",
+  "equinox>=0.12.2",
   "flax",
   "orbax",
   "ott-jax==0.5",