Section 5: 1D Heat Equation Neural Operator using PyTorch