Concise Implementation for Multiple GPUs

https://d2l.ai/chapter_computational-performance/multiple-gpus-concise.html