diff --git a/requirements.txt b/requirements.txt
index ac8fd6cc..a59e460a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,5 +4,5 @@ protobuf>=4, <5
 cloudpickle>=2.0.0
 multiprocess>=0.70.12.2
 cachetools>=5.0.0
-jax[cpu]>=0.4.16, <=0.4.26 # FIXME: Jax 0.4.26+ select perf issue
+jax[cpu]>=0.4.31, <=0.4.31 # FIXME: Jax 0.4.26+ select perf issue
 termcolor>=2.0.0