diff --git a/requirements-dev.txt b/requirements-dev.txt
index 33bcd4da..ff887cc2 100644
--- a/requirements-dev.txt
+++ b/requirements-dev.txt
@@ -1,5 +1,5 @@
 pandas>=1.4.2
-flax<0.10.0
+flax<0.10.3
 scikit-learn<1.6.0
 # for tests
 absl-py>=1.1.0