diff --git a/Project.toml b/Project.toml
index 868f639..cf4c214 100644
--- a/Project.toml
+++ b/Project.toml
@@ -32,6 +32,7 @@ cuDNN = "02a925ec-e4fe-4b08-9a7e-0d78e3d38ccd"
 CUDA = "5"
 ChainRulesCore = "1.16.0 - 1"
 DataFrames = "1.6"
+DifferentialEquations = "7.11 - 7"
 Distributions = "0.25"
 GraphNetCore = "0.3"
 HDF5 = "0.17"
@@ -45,6 +46,7 @@ Printf = "1"
 ProgressMeter = "1.7.0 - 1"
 Random = "1"
 SciMLBase = "2.7.0 - 2"
+SciMLSensitivity = "7.45 - 7"
 Statistics = "1"
 TFRecord = "0.4.2"
 Wandb = "0.5"