The cut in the Mixture of Experts compute graph

The MoE backward pass hides a small trick that, if you miss it, means you'll never train. Let's derive it by hand to see why.

添加评论
点赞收藏
点踩分享查看原文
评论
?
参与讨论