jax

7628bff8 - Skip is_finite tests on ROCm (not in Triton lowering for jax 0.8.0) (#597)