mpi4jax: Zero-copy MPI communication of JAX arrays

Submitted 10 June 2021Published 01 September 2021

Editor: @kellyrowland (all papers)
Reviewers: @1313e (all reviews), @Himscipy (all reviews)


Dion Häfner (0000-0002-4465-7317), Filippo Vicentini


Häfner et al., (2021). mpi4jax: Zero-copy MPI communication of JAX arrays. Journal of Open Source Software, 6(65), 3419,

@article{Häfner2021, doi = {10.21105/joss.03419}, url = {}, year = {2021}, publisher = {The Open Journal}, volume = {6}, number = {65}, pages = {3419}, author = {Dion Häfner and Filippo Vicentini}, title = {mpi4jax: Zero-copy MPI communication of JAX arrays}, journal = {Journal of Open Source Software} }
Copy citation string · Copy BibTeX  

Python JAX MPI high performance computing parallel computing

Markdown badge



Authors of JOSS papers retain copyright.

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

Creative Commons License

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