In this HW, you will implement several neural networks using PyTorch.
Please follow the description in this jupyter notebook