Kabalan, W., & Lanusse, F. (2025). jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs (Version 0.2.9) [Computer software]. https://doi.org/10.5281/zenodo.18512020