Results 2 comments of Qinwen

will waiting for PR approval until merge, but need to tag 'pull ready' to unblock internal testing

@jesselu-google do you want to include some simple moe test to cover this gather/scatter_add?