mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 03:46:06 +00:00

Previously these errors came from Mosaic with less useful stack traces, and in the case of GPU we were getting a crash instead of an exception. PiperOrigin-RevId: 657184114