TikTok interview question

Implement backprop for two layer mlp.