diff --git a/python_jax/requirements.txt b/python_jax/requirements.txt
index 4a153e2..5c604fc 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
+jax>= 0.12.0.1
 flax>= 0.12.10