diff --git a/Project.toml b/Project.toml
index 48418ef4ac..565d20c6b6 100644
--- a/Project.toml
+++ b/Project.toml
@@ -30,7 +30,7 @@ ChainRulesCore = "1.12"
 Functors = "0.3, 0.4"
 MLUtils = "0.2, 0.3.1"
 MacroTools = "0.5"
-NNlib = "0.8.9"
+NNlib = "0.8.9, 0.9"
 NNlibCUDA = "0.2.4"
 OneHotArrays = "0.1, 0.2"
 Optimisers = "0.2.10"