jax

2334ef6d - [Mosaic GPU] Add a `ScopedDiagnosticHandler` to propagate deserialization failures.