Redesign data handling for multi-GPU support

[mchaker]

I attempted to run equibind with multiple GPUs and only one was used.

A quick search found the following solution:
https://stackoverflow.com/questions/54216920/how-to-use-multiple-gpus-in-pytorch

Could the code of this project be fully parallelized (to take advantage of multiple GPUs)?