Cache compiled JAX init HLOs in MaxEngine.#4643

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

mainAI-Hypercomputer/maxtext:mainfrom

test_955227312AI-Hypercomputer/maxtext:test_955227312Copy head branch name to clipboard