jax

Enable testSvdSubsetByIndex for ROCm with subset_by_index skip

#603

Merged