diff --git a/Project.toml b/Project.toml
index 284271e..c47cd26 100644
--- a/Project.toml
+++ b/Project.toml
@@ -20,6 +20,7 @@ FillArrays = "0.8, 0.9, 0.10, 0.11, 0.12, 0.13, 1"
 Flux = "0.11"
 NNlib = "0.6, 0.7.0 - 0.7.24"
 Requires = "1"
+Statistics = "1"
 Torch_jll = "~1.4"
 Zygote = "0.5"
 ZygoteRules = "0"