Add expert choice routing mode to MoEFeedForward#8

ahmedtaha100 wants to merge 2 commits into

google-deepmind:maingoogle-deepmind/simply:mainfrom

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