Optiver人工智能面试真题

Implement a simple neural network with one hidden layer using PyTorch. Include the network definition, optimizer, cost function, and the training process.