Updated scipy and jax versions to fix linalg.tri deprecation

[brosaplanella]

Description

Updated jax & jaxlib to most recent version and scipy to >= 1.11.0 (the one that introduced the deprecation warning).

Fixes #3959

Type of change

Please add a line in the relevant section of CHANGELOG.md to document the change (include PR #) - note reverse order of PR #s. If necessary, also add to the list of breaking changes.

• New feature (non-breaking change which adds functionality)
• Optimization (back-end change that speeds up the code)
• Bug fix (non-breaking change which fixes an issue)

Key checklist:

• No style issues: $ pre-commit run (or $ nox -s pre-commit) (see CONTRIBUTING.md for how to set this up to run automatically when committing locally, in just two lines of code)
• All tests pass: $ python run-tests.py --all (or $ nox -s tests)
• The documentation builds: $ python run-tests.py --doctest (or $ nox -s doctests)

You can run integration tests, unit tests, and doctests together at once, using $ python run-tests.py --quick (or $ nox -s quick).

Further checks:

• Code is commented, particularly in hard-to-understand areas
• Tests added that prove fix is effective or that feature works

[kratman]

@brosaplanella It looks like the remaining failures are about precision

[brosaplanella]

Ok, I found the issue. Basically they have changed the solver for the interpolant and now it is an iterative one (see Scipy docs). I have now defined the interpolant to use a direct solver (as before) so accuracy should remain. If needed, in the future, we can allow users to specify the solver when creating the interpolant.

EDIT: just realised that this new feature is only supported from 1.13 so it would break with older Scipy versions. @valentinsulzer, how should we deal with this? Pin Scipy to 1.13 or is there a way to support both?

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 99.54%. Comparing base (8823b40) to head (5a237ea).
Report is 213 commits behind head on develop.

Additional details and impacted files

@@             Coverage Diff             @@
##           develop    #4103      +/-   ##
===========================================
- Coverage    99.56%   99.54%   -0.02%     
===========================================
  Files          287      287              
  Lines        21744    21744              
===========================================
- Hits         21649    21646       -3     
- Misses          95       98       +3     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[kratman]

@brosaplanella If the only issue was precision before switching to the direct solver, then I would guess that the iterative solver would be fine. If they made that the default, then there is probably a reason for it

[kratman]

Assuming tests pass, then I am happy with this

[agriyakhetarpal]

Thanks for taking the time to do this, @brosaplanella! The changes here were a bit easier than I expected they would be, just one comment on the version bump