diff --git a/requirements.txt b/requirements.txt
index a01ec1c..97495fd 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -18,3 +18,4 @@ ipywidgets<=7.7.1
 pre-commit
 pytest
 jax <= 0.4.23
+jaxlib <= 0.4.23