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 (gemma2_2b_keras_jax_l4_1h1d_presubmit)

succeeded Feb 25, 2026 in 8m 40s