jax

36a03dd4 - [ROCm]: Add support to continue on fail