Meta interview question

Effect of various loop orders on the hardware performance. Small pytorch code making a sequential model