Skip to content

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

Updated flax_gspmd/bridge_guide to use kernel_metadata, make_mesh and sharding_names

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

Annotations

1 warning

Run Tests (3.13, pytest, newest)

succeeded Nov 20, 2025 in 2m 41s