diff --git a/Project.toml b/Project.toml
index 0c86f31..9d93b20 100644
--- a/Project.toml
+++ b/Project.toml
@@ -33,7 +33,7 @@ CUDA = "5.5.2"
 ChainRulesCore = "1.25"
 Combinatorics = "1.0.2"
 ComponentArrays = "0.15.19"
-Flux = "v0.15.2"
+Flux = "v0.15.2, 0.16"
 GPUArraysCore = "0.1, 0.2"
 LinearAlgebra = "1.10.0"
 Lux = "1.4.2"