jax
776f8a90 - Added a release validation workflow for JAX ROCm wheels.

Commit
27 days ago
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)
Author
Committer
Parents
Loading