Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.4.1...v0.4.1.post1
Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.4.0.post6...v0.4.1
Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.4.0.post5...v0.4.0.post6
Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.4.0.post4...v0.4.0.post5
Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.4.0.post3...v0.4.0.post4
Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.4.0.post2...v0.4.0.post3
Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.4.0.post1...v0.4.0.post2
Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.4.0...v0.4.0.post1
This release adds experimental support for mpi4jax
primitives that do not require manual token management, located in mpi4jax.experimental.notoken
. To test the new notoken mode in existing mpi4jax code, use MPI4JAX_PREFER_NOTOKEN=1
.
mpi4jax.experimental.notoken
module added by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/212
mpi4jax.experimental.auto_tokenize
is removed.Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.15...0.4.0
Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.15.post1...v0.3.15.post2