jax

a65d3ae0 - [Mosaic] Expand vector.shape_cast support for sublane (un)folding no-ops

[Mosaic] Expand vector.shape_cast support for sublane (un)folding no-ops - Support non-zero minor offsets without having to relayout (they're still a no-op). - Remove restriction on tiling which now allows 1D packed types to work. PiperOrigin-RevId: 640967375