We are going to try creating a simple neural network using nothing but matrix operation to showcase basic NN functionality. After that, we are going to update our implementation to make better use of the tools offered by PyTorch.

Goals:

• To understand the neural network representation and the optimization process,
• Using `Dataset` and `DataLoader` classes to manipulate training data,
• Learning the basics of the `nn.Module` class and `torch.optim` library,
  
• Switching between 'GPU' and 'CPU' computation.