Source: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.fft.fftshift.html.pdf