Metal Performance Shaders, Apple's framework for GPU-accelerated computing, supported by PyTorch.
Latent Space