Update jax requirement from <=0.4.24 to <=0.4.28

Updates the requirements on [jax](https://github.com/google/jax) to permit the latest version.
- [Release notes](https://github.com/google/jax/releases)
- [Changelog](https://github.com/google/jax/blob/main/CHANGELOG.md)
- [Commits](google/jax@jaxlib-v0.1.32...jax-v0.4.28)

---
updated-dependencies:
- dependency-name: jax
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>

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