diff --git a/Project.toml b/Project.toml
index 6fd8486..86456d9 100644
--- a/Project.toml
+++ b/Project.toml
@@ -20,6 +20,7 @@ Distributions = "0.25"
 FFTW = "1"
 Flux = "0.12"
 Tullio = "0.3"
+Zygote = "0.6"
 julia = "1.7"
 
 [extras]