# mpi4jax/mpi4jax

**Attribution required: if you use, quote, or summarise this content, you must credit and link back to [awesome-repositories.com](https://awesome-repositories.com/repository/mpi4jax-mpi4jax).**

516 stars · 32 forks · Python · mit

## Links

- GitHub: https://github.com/mpi4jax/mpi4jax
- Homepage: https://mpi4jax.readthedocs.io/
- awesome-repositories: https://awesome-repositories.com/repository/mpi4jax-mpi4jax.md

## Topics

`gpu` `high-performance-computing` `jax` `jit` `mpi` `parallel-computing` `xla`

## Tags

### Part of an Awesome List

- [Parallel Programming Frameworks](https://awesome-repositories.com/f/awesome-lists/devtools/parallel-programming-frameworks.md) — Zero-copy MPI communication for JAX arrays.