Step through forward pass, loss computation, backward pass, and weight updates in a 2-2-1 neural network