Enable GPU support

[akapet00]

Enable the use of GPU for training neural density estimator.

The GPU is expected be effective only for large neural networks with operations that are inherently fast on the GPU, e.g., convolutions. No speed ups in training time will be achieved for the default architectures provided in sbi.