Types of changes

This PR fixes the test in the backend requiring that jax version must be more recent than "0.4.24".

Motivation and context / Related issue

Jax changed their numerotation of versions from "0.9.0" to "0.9.0.1", see https://github.com/jax-ml/jax/releases .

How has this been tested (if it applies)

PR checklist

• I have read the CONTRIBUTING document.
• The documentation is up-to-date with the changes I made (check build artifacts).
• All tests passed, and additional code has been covered with new tests.
• I have added the PR and Issue fix to the RELEASES.md file.