jax

bcb3e200 - [pallas:mosaic] Fixed the signature of `infer_tiling` -- it never returns `None`