diff --git a/dependencies/requirements/base_requirements/requirements.txt b/dependencies/requirements/base_requirements/requirements.txt
index b2a5d239c..a446ae034 100644
--- a/dependencies/requirements/base_requirements/requirements.txt
+++ b/dependencies/requirements/base_requirements/requirements.txt
@@ -6,7 +6,7 @@ av
 chex
 datasets
 einops
-flax
+flax>=0.12.10
 ftfy
 google-cloud-storage
 grain
diff --git a/dependencies/requirements/generated_requirements/requirements.txt b/dependencies/requirements/generated_requirements/requirements.txt
index da356c936..0e66b4637 100644
--- a/dependencies/requirements/generated_requirements/requirements.txt
+++ b/dependencies/requirements/generated_requirements/requirements.txt
@@ -41,7 +41,7 @@ etils>=1.13.0
 execnet>=2.1.2
 filelock>=3.20.3
 flatbuffers>=25.12.19
-flax>=0.12.6
+flax>=0.12.10
 fonttools>=4.61.1
 frozenlist>=1.8.0
 fsspec>=2026.1.0
diff --git a/src/maxdiffusion/dependency_versions_table.py b/src/maxdiffusion/dependency_versions_table.py
index 241035ac3..b6074e551 100644
--- a/src/maxdiffusion/dependency_versions_table.py
+++ b/src/maxdiffusion/dependency_versions_table.py
@@ -8,7 +8,7 @@
     "aqtp": "aqtp",
     "datasets": "datasets",
     "einops": "einops",
-    "flax": "flax",
+    "flax": "flax>=0.12.10",
     "ftfy": "ftfy",
     "google-cloud-storage": "google-cloud-storage",
     "grain": "grain",