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.