jax

9518f1df - Strip down workflow to execute only rocm piece.