Enable use of CUDA for training if available

[jchodera]

We should add code to our training notebooks to enable us to use CUDA for training if it is available.

From the espaloma example, we could add something like this:

if torch.cuda.is_available():
    espaloma_model = espaloma_model.cuda()