diff --git a/python_jax/requirements.txt b/python_jax/requirements.txt
index 4a153e2..b4d5a59 100644
--- a/python_jax/requirements.txt
+++ b/python_jax/requirements.txt
@@ -1,4 +1,4 @@
 numpy>= 2.5.4
 opencv-python
 jax>= 0.12.0
-flax>= 0.12.10
+flax>= 0.12.12