flax

3e963438 - Rename sharding_names to sharding_metadata