simple multi-gpu support

[theRealSuperMario]

the related project pytorch-lightning implements an easy way to just scale the project onto multiple GPUs. I think we should adapt it somehow, at least when working on a single machine.