Further improvements to multi-GPU scaling