diff --git a/Project.toml b/Project.toml
index 0fb43b8..a65864f 100644
--- a/Project.toml
+++ b/Project.toml
@@ -23,7 +23,7 @@ ChainRules = "1.72.6"
 ChainRulesCore = "1.25.2"
 DiffOpt = "0.5.4"
 FillArrays = "1.14.0"
-Flux = "0.14.21"
+Flux = "0.14.21, 0.16"
 JLD2 = "0.6.3"
 JuMP = "1.29.4"
 MathOptInterface = "1.48.0"