diff --git a/Project.toml b/Project.toml
index 47f8a8a..2a96ab3 100644
--- a/Project.toml
+++ b/Project.toml
@@ -20,6 +20,7 @@ StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
 Adapt = "3"
 Distributions = "0.25"
 InverseFunctions = "0.1"
+MLUtils = "0.4"
 NNlib = "0.8"
 OneHotArrays = "0.2"
 Quaternions = "0.7"