diff --git a/Project.toml b/Project.toml
index bf730f53..06dccfc0 100644
--- a/Project.toml
+++ b/Project.toml
@@ -5,8 +5,8 @@ version = "0.9.0"
 [deps]
 Artifacts = "56f22d72-fd6d-98f1-02f0-08ddc0907c33"
 BSON = "fbb218c0-5317-5bc6-957e-2ee96dd4b1f0"
-ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
+ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
 Functors = "d9f16b24-f501-4c13-a1f2-28368ffc5196"
 JLD2 = "033835bb-8acc-5ee8-8aae-3f567f8a3819"
@@ -33,4 +33,5 @@ JLD2 = "0.4"
 MLUtils = "0.4"
 NNlib = "0.8, 0.9"
 PartialFunctions = "1"
+Statistics = "1"
 julia = "1.6"