diff --git a/Project.toml b/Project.toml
index dab75fd..b3bf9cf 100644
--- a/Project.toml
+++ b/Project.toml
@@ -50,6 +50,7 @@ FastClosures = "0.3"
 FiniteDiff = "2.12"
 ForwardDiff = "0.10.26"
 LinearAlgebra = "1.9, 1.10"
+MLDataDevices = "1"
 ModelingToolkit = "9"
 PDMats = "0.11"
 Reexport = "1.2"