jax

833c7ba7 - Allow generation of sharding strategies with mixed mesh shapes by default.