Update minimum jax/jaxlib versions

[ColmTalbot]

I wasn't sure if I should bump both jax and jaxlib?

@nstarman

Closes #237

[nstarman]

Thanks! In tandem is good.

[codecov[bot]]

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 96.17%. Comparing base (2554aeb) to head (49c36f6). Report is 1 commits behind head on main.

Additional details and impacted files

```diff @@ Coverage Diff @@ ## main #238 +/- ## ======================================= Coverage 96.17% 96.17% ======================================= Files 41 41 Lines 1621 1621 ======================================= Hits 1559 1559 Misses 62 62 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.

[nstarman]

Thanks!