Walmart Global Tech interview question

2 rounds mainly general ML - code LR from scratch - depth questions about transformers