can be used on GPUs via PyTorch