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

@article{Häfner2021, doi = {10.21105/joss.03419}, url = {https://doi.org/10.21105/joss.03419}, 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  
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

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