TikTok interview question

implement forward and backward training loop logistic regression from scratch.