v0.7.0
What's Changed
- Drop support for Python<3.10.
- Drop support for JAX<0.4.35.
- New public module
mpi4jax.notoken
. Removedmpi4jax.experimental.notoken
. - All primitives now use "notoken" mode under the hood (can be disabled via
MPI4JAX_PREFER_NOTOKEN=0
).
Please report any issues with these changes.
Full Changelog: v0.6.1...v0.7.0