diff --git a/requirements_mtj.txt b/requirements_mtj.txt
index a05c333a..32836175 100644
--- a/requirements_mtj.txt
+++ b/requirements_mtj.txt
@@ -5,6 +5,7 @@ requests
 optax >= 0.0.5, <= 0.0.9
 dm-haiku == 0.0.5
 jax == 0.2.21
+jaxlib >= 0.1.69, <= 0.3.7
 transformers >= 4.19
 progressbar2
 git+https://github.com/VE-FORBRYDERNE/mesh-transformer-jax@ck
@@ -17,4 +18,4 @@ lupa==1.10
 markdown
 bleach==4.1.0
 python-socketio[client]
-flask_session
\ No newline at end of file
+flask_session