我认为您正在寻找mpi4py-fft，这是一个 Python 包（BSD-2 许可），其包装器位于串行 FFTW 库中。从相当广泛的mpi4py 文档中：

并行 FFT 是通过全局重新分布和串行变换的组合来计算的。

该库还依赖于mpi4py（也由 mpi4py-fft 的合著者之一开发）。

纸：

• L. Dalcin、M. Mortensen 和 DE Keyes，“使用高级 MPI 的快速并行多维 FFT”，J. 并行分布式比较。, 卷。128，第 137-150 页，2019 年 6 月。