add use_cuda parameter

[mjfox3]

using the package in an environment without cuda support causes it to fail. Adding the parameter to shut it off if necessary allows it to function normall.

[amiyamandal-dev]

@mjfox3 I guess

use_cuda: bool = torch.cuda.is_available()

might be a better approach