jax

822bc364 - Update quickstart to suggested jax[cuda12] installation

Update quickstart to suggested jax[cuda12] installation This PR updates the quickstart docs to match the README and Installation Guide suggestion of using ``` pip install jax[cuda12] ``` for GPU.