diff --git a/Project.toml b/Project.toml
index 42b9547..4e59372 100644
--- a/Project.toml
+++ b/Project.toml
@@ -23,16 +23,17 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 [compat]
 Arrow = "2"
 CSV = "0.10"
-JuMP = "1"
-ParametricOptInterface = "0.8"
-Zygote = "0.6.68"
-julia = "1.9"
 DataFrames = "1"
+Distributions = "0.25"
 Flux = "0.14"
-Optimisers = "0.3"
+JuMP = "1"
 MLJFlux = "0.6"
-Distributions = "0.25"
 NNlib = "0.9"
+Optimisers = "0.3"
+ParametricOptInterface = "0.8"
+Statistics = "1"
+Zygote = "0.6.68"
+julia = "1.9"
 
 [extras]
 CUDA_Runtime_jll = "76a88914-d11a-5bdc-97e0-2f5a05c973a2"