Fix GPT3 attention missing KV cache initialization and handling#3927

copybara-service[bot] merged 1 commit into

mainAI-Hypercomputer/maxtext:mainfrom

shralex_test_5AI-Hypercomputer/maxtext:shralex_test_5Copy head branch name to clipboard