parallelized stoch_grad_mom