Intro to PyTorch: Training your first neural network using PyTorch