jax

56321132 - Enable polynomial roots tests on ROCm platform on v0.8.0 branch (#665)