if self.vs is None:
            self.vs = [_zero(p.shape) for p in params]