jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs

Python Cuda Submitted 25 June 2025Published 16 May 2026
Review

Editor: @diehlpk (all papers)
Reviewers: @rafmudaf (all reviews), @k20shores (all reviews)

Authors

Wassim Kabalan (0009-0001-6501-4564), François Lanusse (0000-0001-7956-0542), Alexandre Boucaud, Eric Aubourg

Citation

Kabalan et al., (2026). jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs. Journal of Open Source Software, 11(121), 8852, https://doi.org/10.21105/joss.08852

@article{Kabalan2026, doi = {10.21105/joss.08852}, url = {https://doi.org/10.21105/joss.08852}, year = {2026}, publisher = {The Open Journal}, volume = {11}, number = {121}, pages = {8852}, author = {Kabalan, Wassim and Lanusse, François and Boucaud, Alexandre and Aubourg, Eric}, title = {jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs}, journal = {Journal of Open Source Software} }
Copy citation string · Copy BibTeX  
Tags

Jax CUDA HPC FFT Simulations

Altmetrics
Markdown badge

 

License

Authors of JOSS papers retain copyright.

This work is licensed under a Creative Commons Attribution 4.0 International License.

Creative Commons License

Table of Contents
Public user content licensed CC BY 4.0 unless otherwise specified.
ISSN 2475-9066