jax

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