diff --git a/Project.toml b/Project.toml
index 2e092d4..f272545 100644
--- a/Project.toml
+++ b/Project.toml
@@ -21,6 +21,7 @@ FunctorsExt = "Functors"
 ChainRulesCore = "1"
 Functors = "0.4"
 LinearAlgebra = "1"
+MLUtils = "0.4"
 NNlib = "0.8, 0.9"
 Random = "1"
 julia = "1.9"