jax

d161e1ba - Add text to documentation and to jaxlib version check suggesting a pip update. (#1506)

Add text to documentation and to jaxlib version check suggesting a pip update. (#1506) * Add instructions to update `pip` to README.md. * Add text that suggests upgrading pip if we see jaxlib 0.1.23. * Remove extra text from README.md.