jax
Enable optimizer tests on ROCm
#631
Merged

Loading