diff --git a/Project.toml b/Project.toml
index 609cd6a..6bf45af 100644
--- a/Project.toml
+++ b/Project.toml
@@ -26,7 +26,7 @@ CategoricalDistributions = "0.1.15"
 ChainRulesCore = "1.23.0"
 Compat = "4.7.0"
 Distributions = "0.25.109"
-Flux = "0.12, 0.13, 0.14"
+Flux = "0.12, 0.13, 0.14, 0.16"
 LinearAlgebra = "1.7, 1.10"
 MLJBase = "1"
 MLJModelInterface = "1.8.0"