diff --git a/pyproject.toml b/pyproject.toml
index a378ad3..db29845 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -22,10 +22,12 @@ dependencies = [
     "absl-py",
     "attrs",
     "einops",
+    "fancyflags",
     "jax",
     "numpy",
     "scipy",
     "six",
+    "swirl-dynamics@git+https://github.com/google-research/swirl-dynamics",
     "tensorflow",
 ]