[Issue]: Release on PyPi

[PhilipVinc]

Problem Description

The most recent version I can find on pypi is jaxlib 0.7.1 at https://pypi.org/project/jax-rocm7-plugin/

However it seems you have release version 0.8 on GitHub, but it's not available on pypi.

Could you please publish the wheels on pypi as well?

[lucbruni-amd]

Thanks for reporting, Filippo. @Ruturaj4, is there a plan for this? This would also align the version constraints enforced by JAX's setup.py and would allow users to install from PyPI with, for example, pip install jax[rocm]==0.8.2, as was the concern in #178.