Skip to content

Support rank for replicated sharding during NamedSharding -> SdySharding conversion in order to preserve optimizations like OpenWhileFreeVarsSharding. #53517

Support rank for replicated sharding during NamedSharding -> SdySharding conversion in order to preserve optimizations like OpenWhileFreeVarsSharding.

Support rank for replicated sharding during NamedSharding -> SdySharding conversion in order to preserve optimizations like OpenWhileFreeVarsSharding. #53517

Annotations

3 warnings

Run Benchmark (gemma3_1b_flax_call_x86_1h1d_presubmit)

succeeded Feb 25, 2026 in 3m 46s