jax

8188c574 - [Mosaic:TPU][NFC] Small cleanup of extui rule in apply-vector-layout

[Mosaic:TPU][NFC] Small cleanup of extui rule in apply-vector-layout Removed some duplicate variables, changed dyn_cast to cast, and used in/out consistently instead of source/dst PiperOrigin-RevId: 707836363