flax
Rename sharding_names to sharding_metadata
#5089
Merged

Loading