Understanding PyTorch for AI