jax

3d72c19b - Remove outdated ROCm skip that depends on ROCm versions < 6.4 (#776)