jax

3e1e9899 - [Mosaic] Handle adding singleton minor dimension that was already implicit for non-32-bit types, and do not force native tiling

[Mosaic] Handle adding singleton minor dimension that was already implicit for non-32-bit types, and do not force native tiling Also fix extra comma in apply_vector_layout_test which was being annoying with autoformatter PiperOrigin-RevId: 642454594