[Draft] [GPU] DSv4 Support for TE MoE Block - #5487

[Draft] [GPU] DSv4 Support for TE MoE Block#5487

dinodeep wants to merge 1 commit into

AI-Hypercomputer:mainAI-Hypercomputer/maxtext:mainfrom

dinodeep:dinodeep/te-moe-block-dsv4-supportdinodeep/maxtext:dinodeep/te-moe-block-dsv4-supportCopy head branch name to clipboard