new_mps[:, :self.underlying_model.n_modelparams] = self._inv_transform(
                    new_mps[:, :self.underlying_model.n_modelparams]