We should add code to our training notebooks to enable us to use CUDA for training if it is available. From the [espaloma example](https://espaloma.wangyq.net/experiments/qm_fitting.html), we could add something like this: ```python if torch.cuda.is_available(): espaloma_model = espaloma_model.cuda() ```
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: