diff --git a/Project.toml b/Project.toml
index 0da4eaf..ddc1f28 100644
--- a/Project.toml
+++ b/Project.toml
@@ -18,7 +18,7 @@ Dagger = "0.14.4"
 DaggerGPU = "0.1"
 Flux = "0.12, 0.13"
 Functors = "0.2"
-NNlib = "0.7, 0.8"
+NNlib = "0.7, 0.8, 0.9"
 Zygote = "0.6"
 julia = "1.6"