TikTok Interview Question

Implement backprop for two layer mlp.