Skip to content

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

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. #53155

Annotations

3 warnings

Run Benchmark (gemma3_1b_flax_call_x86_1h1d_presubmit)

succeeded Feb 23, 2026 in 2m 53s