jax

e4f757d0 - [Mosaic GPU] Check collective attribute is consistent across tcgen05 operations.

[Mosaic GPU] Check collective attribute is consistent across tcgen05 operations. It is an error to mix collective and non-collective operations in the same kernel. PiperOrigin-RevId: 887978639