Add support for ROCm wheel based install#305

mrodden wants to merge 1 commit intorocm-jaxlib-v0.4.35-qaROCm/jax:rocm-jaxlib-v0.4.35-qafrom

rocm-wheels-0.4.35ROCm/jax:rocm-wheels-0.4.35Copy head branch name to clipboard