param only can be "mu" or "sigma2" because this is a normal distribution, which can be specific by mu and sigma2.
Here, we take the derivative wrt to mu or sigma2