a Python package for computing Fast Fourier Transforms (FFTs) with MPI

 mpi4py-fft is a Python package for computing Fast Fourier Transforms
 (FFTs). Large arrays are distributed and communications are handled
 under the hood by MPI for Python (mpi4py). To distribute large arrays
 we are using a new and completely generic algorithm that allows for
 any index set of a multidimensional array to be distributed. We can
 distribute just one index (a slab decomposition), two index sets
 (pencil decomposition) or even more for higher-dimensional arrays.
 .
 In mpi4py-fft there is also included a Python interface to the FFTW
 library. This interface can be used without MPI, much like pyfftw,
 and even for real-to-real transforms, like discrete cosine or sine
 transforms.
 .
 The package provides a Python interface to FFTW, but with MPI
 parallelisation. This enables strong scaling tested to 16000 cores,
 or weak scaling tested to 2000 cores. The algorithm is documented at
 https://arxiv.org/abs/1804.09536
 .
 This package installs the library for Python 3.