SHIFT is a scalable interface library for computing FFTs with scipy. The library wraps scipy FFT routines, making it easy to keep track of the corresponding Fourier modes in Fourier space. The package can be used for very large computations with MPI distributed code possible using the mpi4py library, using a slab decomposition to perform distributed FFTs.
NOTE: SHIFT was originally designed to be a Spherical/Polar Fourier Transform library. However, the package has developed into being predominanty a helper and MPI interface for FFTs in 1D/2D/3D cartesian grids. The development of the Polar and Spherical Bessel transforms is still ongoing but is no longer the focus -- functions in
shift.polarandshift.sphereshould be used with extreme caution and are only available from thespherebranch of the code.
numbanumpyscipympi4py[Optional: enables MPI distributed processes]
SHIFT can be installed via pip:
pip install shift-fft
Or by cloning the github repository:
git clone https://github.com/knaidoo29/SHIFT.git
cd SHIFT
pip install .
The latter of which can be used to install specific branches.
Once this is done you should be able to call SHIFT from python:
import shiftTo use the MPI functionality please take a look at the documentation which instructs users how to use the SHIFT MPI object and how to run these distributed jobs successfully without errors or MPI related hanging.
In depth documentation and tutorials are provided here.
If you use SHIFT in your work, please cite:
@software{naidoo_shift_2025,
author = {Naidoo, Krishna},
title = {SHIFT: a scalable MPI library for computing fast Fourier transforms in python},
year = 2025,
publisher = {Zenodo},
doi = {10.5281/zenodo.17093446},
url = {https://doi.org/10.5281/zenodo.17093446}}If you have any issues with the code or want to suggest ways to improve it please open a new issue (here) or (if you don't have a github account) email krishna.naidoo.11@ucl.ac.uk.