jax

4a14e0c2 - [mosaic] Verify that `tpu.memref_slice` always produces tile-aligned slices