def METHOD_NAME(self, x: Tensor) -> MultivariateNormal:
    if self.training:
        x = self.transform_inputs(x)
    mean_x = self.mean_module(x)
    covar_x = self.covar_module(x)
    return MultivariateNormal(mean_x, covar_x)