TODO: make dataparallel work over many GPUs

[JulianKnodt]

I've not really had the chance to check if DataParallel works for multiple GPUs, so getting that to work would be fantastic.