diff --git a/Project.toml b/Project.toml
index 6230fdbee..691003944 100644
--- a/Project.toml
+++ b/Project.toml
@@ -20,9 +20,13 @@ Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 [compat]
 BSON = "0.3.2"
 Flux = "0.13"
-Functors = "0.2"
-MLUtils = "0.2.6"
-NNlib = "0.7.34, 0.8"
+Functors = "0.2, 0.3"
+CUDA = "3"
+ChainRulesCore = "1"
+PartialFunctions = "1"
+MLUtils = "0.2.10"
+NNlib = "0.8"
+NNlibCUDA = "0.2"
 julia = "1.6"
 
 [publish]