Add expert choice routing mode to MoEFeedForward#8

ahmedtaha100 wants to merge 1 commit intogoogle-deepmind:maingoogle-deepmind/simply:mainfrom

ahmedtaha100:expert_choice_routingahmedtaha100/simply:expert_choice_routingCopy head branch name to clipboard