Updated flax_gspmd/bridge_guide to use kernel_metadata, make_mesh and sharding_names#5088

vfdev-5 wants to merge 3 commits intomaingoogle/flax:mainfrom

use-kernel-metadata-in-gspmd-guidegoogle/flax:use-kernel-metadata-in-gspmd-guideCopy head branch name to clipboard