jax

Skip is_finite tests on ROCm (not in Triton lowering for jax 0.8.0)

#597

Merged