GPU Support for Mac Users

[ramosv]

Mac now support Accelerated PyTorch training on Mac via their M-series chips see: https://developer.apple.com/metal/pytorch/

Would be nice for mac users to leverage their GPU for accelerated training. I will be working on adding this before v1.3.0 release

[ramosv]

Example of what would need to be added: The middle elif is all we need to add to the existing block

if torch.cuda.is_available():
    device = torch.device("cuda")
elif torch.backends.mps.is_available():
    device = torch.device("mps")
else:
    device = torch.device("cpu")