NVIDIA Interview Question

Construct Feedforward neural network training loop.