Optimization under uncertainty using momentum