JaxDecomp Documentation
=======================

A high-performance, differentiable distributed FFT library for JAX.


User Guide
==========

.. toctree::
   :maxdepth: 2
   :caption: User Guide

   installation
   01-basic_usage
   02-decomposition
   03-distributed_ffts
   04-halo_exchange
   05-transpositions
   06-caveats
   07-xla_sharding_configuration

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

.. toctree::
   :maxdepth: 2
   :caption: API Reference

   api