Update link to JAX installation page

[SupreethRao99]

The original link to the JAX installation README was broken, it resulted in a 404 page not found error, so I replaced it with the new link (https://github.com/google/jax#installation)

Fixes #2128

Checklist

• This PR fixes a minor issue (e.g.: typo or small bug) or improves the docs (you can dismiss the other
  checks if that's the case).
• This change is discussed in a Github issue/
  discussion (please add a
  link).
• The documentation and docstrings adhere to the
  documentation guidelines.
• This change includes necessary high-coverage tests.
  (No quality testing = no merge!)

[marcvanzee]

Thank you!

[codecov-commenter]

Codecov Report

Merging #2129 (4c8e278) into main (5258eb0) will not change coverage.
The diff coverage is n/a.

@@           Coverage Diff           @@
##             main    #2129   +/-   ##
=======================================
  Coverage   74.97%   74.97%           
=======================================
  Files          59       59           
  Lines        5062     5062           
=======================================
  Hits         3795     3795           
  Misses       1267     1267           

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update 5258eb0...4c8e278. Read the comment docs.