Recommend installing the PyPI released package first, instead of installing from GitHub, as it may not be as stable as the released version.
Recommend installing the PyPI released package first, instead of installing from GitHub, as it may not be as stable as the released version.

Fix the GitHub URL to use `git+https://github.com/...` instead of `git+git://....`

PiperOrigin-RevId: 462700362

deepmind/optax,deepmind/optax