jax

1e36cbe5 - [Mosaic GPU] Raise a `NotImplementedError` if `swizzle=16`.

[Mosaic GPU] Raise a `NotImplementedError` if `swizzle=16`. Unswizzled MMAs don't lower correctly, and are not currently intended to be supported. PiperOrigin-RevId: 737981373