2B) 

Hi there! Thanks for putting this together. Q - shouldnt this equation include a sum over heads term outside the m (MLP) function as well?