Added a release validation workflow for JAX ROCm wheels.
The workflow allows the user to specify an existing JAX release
and will build the appropriate jax-rocm-plugin and jax-rocm-pjrt
wheels. These wheels are then tested on legacy ROCm and TheRock.
If no JAX release is specified, the workflow will build ROCm plugin
and PJRT wheels for the latest JAX release. Suffix naming for post
releases is also handled automatically.
(cherry picked from commit de0371637531f4a138f95895638db24256ee65cd)