diff --git a/Project.toml b/Project.toml
index cf5a675..28d6e75 100644
--- a/Project.toml
+++ b/Project.toml
@@ -26,7 +26,7 @@ ArrayInterface = "6, 7"
 DiffEqBase = "6.123.0"
 FiniteDiff = "2"
 ForwardDiff = "0.10.3"
-NNlib = "0.8"
+NNlib = "0.8, 0.9"
 Reexport = "0.2, 1"
 Requires = "1"
 SciMLBase = "1.73"