Train Pytorch with GPU on Apple Silicon (M1 series)