To anyone stumbling upon this thread - VkFFT now supports Metal API to do FFTs on GPUs: https://github.com/DTolm/VkFFT