mpi4jax: Zero-copy MPI communication of JAX arrays

Submitted 10 June 2021Published 01 September 2021
Review

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

Authors

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

Citation

Häfner et al., (2021). mpi4jax: Zero-copy MPI communication of JAX arrays. Journal of Open Source Software, 6(65), 3419, https://doi.org/10.21105/joss.03419

Copy citation string · Copy BibTeX  
Tags

Python JAX MPI high performance computing parallel computing

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

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