Remove jax version constraint when pymanopt use new jax config format

[rflamary]

Describe the bug

We needed to force an older version of jax to prevent a bug when importing pymanopt because of a very recent deprecation of the configuration setup of jax. We need to remove this when

This has been done in #608 but should be reversed when possible.