diff --git a/pyproject.toml b/pyproject.toml
index 07116d8..358d26a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -16,7 +16,7 @@ keywords = []
 # pip dependencies of the project
 dependencies = [
     "jax==0.4.35",
-    "flax==0.10.1",
+    "flax==0.10.2",
     "ml_dtypes==0.4.0",
     "optax==0.2.4",
     "orbax-checkpoint==0.9.1",