Fix: Gemma 3 & 4 Base Model and Flax Linen Decoding - #4066

Fix: Gemma 3 & 4 Base Model and Flax Linen Decoding #4066

copybara-service[bot] merged 1 commit into

mainAI-Hypercomputer/maxtext:mainfrom

jackyf/gemma3_4-base-decoding-fixesAI-Hypercomputer/maxtext:jackyf/gemma3_4-base-decoding-fixesCopy head branch name to clipboard