BGD AllReduce with chunked data