Use gather/broadcast with trees for cluster comm.