I was asked to construct a Feedforward neural network training loop on the spot -- not necessarily coded, but asked to describe the overall architecture with mock data. I could use either pytorch or tensorflow