jax

368b60e1 - Strip down workflow to execute only rocm piece.