Update jax requirement from <=0.4.14 to <=0.4.25 in /tests

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](jax-ml/jax@jaxlib-v0.1.32...jax-v0.4.25)

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

Signed-off-by: dependabot[bot] <[email protected]>

tests/constraints.txt

@@ -22,6 +22,6 @@ pytest>=5.0,<7.5
 pytest-dependency==0.5.1
 pytest-html==4.1.1
 pytest-timeout==2.2.0
-jax<=0.4.14
+jax<=0.4.25
 jaxlib<=0.4.14
 torch>=1.13,<2.3