CS336 Lecture 4 - MoE