Auto-layout ahead of time compilation for JetStream/Maxtext#259

jwyang-google wants to merge 1 commit intomainAI-Hypercomputer/JetStream:mainfrom

jetstream_aotAI-Hypercomputer/JetStream:jetstream_aotCopy head branch name to clipboard