flax

Updated flax_gspmd/bridge_guide to use kernel_metadata, make_mesh and sharding_names

#5088

Open