Update README.md so that CPU install instruction works
Fix readme
current `pip install -U jax[cpu]` has no matches in pip `no matches found: jax[cpu]`.
Corrected install instruction to `pip install -U "jax[cpu]"` which successfully installs cpu version of JAX via pip.
