[MRG] Temporary limit on jax version

[rflamary]

Limit the max version of jax to allow for old config because of pymanopt using deprecated version

Types of changes

Motivation and context / Related issue

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.

[codecov]

Codecov Report

Merging #608 (e7688a0) into master (6f35804) will decrease coverage by 0.02%.
The diff coverage is n/a.

Additional details and impacted files

@@            Coverage Diff             @@
##           master     #608      +/-   ##
==========================================
- Coverage   96.78%   96.77%   -0.02%     
==========================================
  Files          77       77              
  Lines       16002    16002              
==========================================
- Hits        15488    15486       -2     
- Misses        514      516       +2