Neural Networks in PyTorch