diff --git a/tests/requirements.txt b/tests/requirements.txt
index 8ad9777..f8b7e3e 100644
--- a/tests/requirements.txt
+++ b/tests/requirements.txt
@@ -11,5 +11,5 @@ torchaudio
 mace-torch
 newtonnet @ git+https://github.com/THGLab/NewtonNet.git
 sella==2.3.4
-jax<=0.4.24
+jax<=0.4.28
 jaxlib<=0.4.24