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