API Reference
=============

FFT
-----------------

.. automodule:: jaxdecomp.fft
   :members:


Halo Exchange
-----------------
.. automodule:: jaxdecomp.halo
   :members:

Transpositions
-----------------
.. automodule:: jaxdecomp.transpose
   :members: