diff --git a/Project.toml b/Project.toml
index 51468e9..63adf98 100644
--- a/Project.toml
+++ b/Project.toml
@@ -20,7 +20,7 @@ StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
 Adapt = "3"
 Distributions = "0.25"
 InverseFunctions = "0.1"
-NNlib = "0.8"
+NNlib = "0.8, 0.9"
 OneHotArrays = "0.2"
 Quaternions = "0.7"
 Rotations = "1.4"