Onboard explicit sharding to deepseek#2579

NuojCheng wants to merge 1 commit intomainAI-Hypercomputer/maxtext:mainfrom

chengnuojin-explicit-deepseekAI-Hypercomputer/maxtext:chengnuojin-explicit-deepseekCopy head branch name to clipboard