diff --git a/requirements.txt b/requirements.txt
index 5a96fef3..695a2f1f 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
+jax[cpu]>=0.4.16, <=0.4.26 # FIXME
 termcolor>=2.0.0