mpi4jax

Community

MPI support for JAX

Versions

To install this package, run one of the following:

$conda install conda-forge::mpi4jax

Version

5 / 8 versions selected

Downloads (Last 6 months): 0

Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python ⚡

Summary

MPI support for JAX

Last Updated

Apr 21, 2026 at 14:46

License

MIT

Supported Platforms

macOS-arm64

linux-64

Unsupported Platforms

macOS-64 Last supported version: 0.7.0

Home

GitHub Repository

Documentation