PyTorch App: Sum of squares = 14.0