Hi, Is it possible for you to upgrade the code using pytorch fft. As you own this code, it will be easy for you to do so. Regards,