diff --git a/Cargo.lock b/Cargo.lock
index 09ede1d..c21323d 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -137,7 +137,7 @@ dependencies = [
  "hex",
  "insta",
  "miette",
- "ndarray",
+ "ndarray 0.17.2",
  "ort",
  "predicates",
  "regex",
@@ -955,6 +955,21 @@ dependencies = [
  "rawpointer",
 ]
 
+[[package]]
+name = "ndarray"
+version = "0.17.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "520080814a7a6b4a6e9070823bb24b4531daac8c4627e08ba5de8c5ef2f2752d"
+dependencies = [
+ "matrixmultiply",
+ "num-complex",
+ "num-integer",
+ "num-traits",
+ "portable-atomic",
+ "portable-atomic-util",
+ "rawpointer",
+]
+
 [[package]]
 name = "nom"
 version = "7.1.3"
@@ -1041,7 +1056,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "1fa7e49bd669d32d7bc2a15ec540a527e7764aec722a45467814005725bcd721"
 dependencies = [
  "libloading",
- "ndarray",
+ "ndarray 0.16.1",
  "ort-sys",
  "smallvec 2.0.0-alpha.10",
  "tracing",
diff --git a/Cargo.toml b/Cargo.toml
index be701af..31509e3 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -45,7 +45,7 @@ sqlite-vec = "0.1"
 zerocopy = { version = "0.8", features = ["derive"] }
 ort = { version = "=2.0.0-rc.10", default-features = false, features = ["load-dynamic"] }
 tokenizers = { version = "0.21", default-features = false, features = ["fancy-regex"] }
-ndarray = "0.16"
+ndarray = "0.17"
 dirs = "6"
 
 [dev-dependencies]