Enable Training on Apple Silicon Processors in PyTorch