flax
Canonicalize PartitionSpec so that we can delete ParsedPartitionSpec
#4577
Open
Open