jax

d667dee4 - [Mosaic GPU] Add layout inference rule for `vector.splat` in the equational layout inference.