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