Support partial overrides for logical_axis_rules.#730

copybara-service[bot] merged 1 commit into

mainAI-Hypercomputer/maxtext:mainfrom

smarter-sharding-parsingAI-Hypercomputer/maxtext:smarter-sharding-parsingCopy head branch name to clipboard