diff --git a/Project.toml b/Project.toml
index 4213b92..2cf117b 100644
--- a/Project.toml
+++ b/Project.toml
@@ -25,17 +25,22 @@ StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [compat]
+AbstractTrees = "0.3"
+Adapt = "2"
 BSON = "0.2"
+CUDA = "1"
 Distributions = "0.22, 0.23"
 FillArrays = "0.8"
 Flux = "0.11"
+GPUArrays = "5"
 ImageTransformations = "0.8"
 JLD = "0.10"
 MacroTools = "0.5"
 ProgressMeter = "1.2"
 ReinforcementLearningBase = "0.8"
-Setfield = "0.6"
+Setfield = "0.6, 0.7"
 StatsBase = "0.32, 0.33"
+Zygote = "0.5"
 julia = "1.3"
 
 [extras]