If you happen to need any assistance in refining VkFFT for your use case, feel free to contact me.