jax

0011fb97 - The ROCm plugin should fail gracefully and allow for other devices to… (#550)