Testing...
parallel_reduce to compute the gradient