diff --git a/Project.toml b/Project.toml
index 4213b92..2ab1e8b 100644
--- a/Project.toml
+++ b/Project.toml
@@ -36,6 +36,7 @@ ProgressMeter = "1.2"
 ReinforcementLearningBase = "0.8"
 Setfield = "0.6"
 StatsBase = "0.32, 0.33"
+Zygote = "0.5"
 julia = "1.3"
 
 [extras]