Enable export-time serialization of XLA GPU flags for Orbax models to natively expose memory optimization features like Latency Hiding Scheduler and Shard Autotuning.#2928

Open

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

Commits on Feb 23, 2026

Enable export-time serialization of XLA GPU flags for Orbax models to natively expose memory optimization features like Latency Hiding Scheduler and Shard Autotuning.
Orbax Authors
committed