Computational Physics - Learning PyTorch