Allow restore+broadcast logic to not require a single-replica sharding parameter, which is always constructed as a sharding over replica-local devices anyway.#3132

Closed

copybara-service[bot] wants to merge 1 commit intomainfrom

Commits on Feb 13, 2026

Allow restore+broadcast logic to not require a single-replica sharding parameter, which is always constructed as a sharding over replica-local devices anyway.

cpgaffney1
authored and
Google-ML-Automation
committed