Support gemma4 decoder under explicit sharding mode - #4766

Support gemma4 decoder under explicit sharding mode#4766

csgoogle wants to merge 1 commit into

AI-Hypercomputer:mainAI-Hypercomputer/maxtext:mainfrom

csgoogle:gemma4-explicit-shardingcsgoogle/maxtext:gemma4-explicit-shardingCopy head branch name to clipboard