Add support for jax.sharding.PartitionSpec.UNCONSTRAINED
in logical specification
#12770
Job | Run time |
---|---|
6s | |
17s | |
5s | |
30s | |
30s | |
26s | |
0s | |
1m 54s |