NVIDIA interview question

Construct Feedforward neural network training loop.