diff --git a/Cargo.toml b/Cargo.toml
index cf82f97..37a92d2 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -67,7 +67,7 @@ netlink-sys = "0.8"
 
 # ML (optional)
 candle-core = { version = "0.3", optional = true }
-candle-nn = { version = "0.3", optional = true }
+candle-nn = { version = "0.10", optional = true }
 
 [features]
 default = []