Pre-release of jaxDecomp, a JAX library providing bindings to NVIDIA’s cuDecomp adaptive pencil decomposition library for efficient multi-node/multi-gpu 3D FFTs and halo exchanges.