What's Changed

• Bump jax from 0.6.2 to 0.7.0 in /mpi4jax/_src by @dependabot[bot] in #283

Full Changelog: v0.8.0...v0.8.1

⚠️ This is an API breaking release ⚠️

Token handling is now removed from mpi4jax. That is, communication primitives do not accept / return token objects anymore, and order of execution is enforced internally by JAX.

To upgrade, please adjust your code like this:

# BEFORE (mpi4jax<0.8.0)
result, token = mpi4jax.allgather(x, token=token)

# AFTER (mpi4jax>=0.8.0)
result = mpi4jax.allgather(x)

What's Changed

• Remove token mode by @dionhaefner in #279

Full Changelog: v0.7.2...v0.8.0

This is the final release that supports passing / returning tokens explicitly.

What's Changed

• Add python-version file to fix dependabot by @dionhaefner in #281
• Bump jax from 0.5.2 to 0.6.2 in /mpi4jax/_src by @dependabot in #282

Full Changelog: v0.7.1...v0.7.2

This is the last release that supports token mode and JAX versions < 0.5.1.

What's Changed

• Fix default no token mode for older JAX versions by @macekmar in #278

New Contributors

• @macekmar made their first contribution in #278

Full Changelog: v0.7.0...v0.7.1

What's Changed

• Bump jax from 0.5.0 to 0.5.1 in /mpi4jax/_src by @dependabot in #276
• Bump jax from 0.5.1 to 0.5.2 in /mpi4jax/_src by @dependabot in #277

Full Changelog: v0.7.0...v0.7.0.post1

What's Changed

• Drop support for Python<3.10.
• Drop support for JAX<0.4.35.
• New public module mpi4jax.notoken. Removed mpi4jax.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

What's Changed

• Bump jax from 0.4.37 to 0.4.38 in /mpi4jax/_src by @dependabot in #272

Full Changelog: v0.6.1.post2...v0.6.1.post3

What's Changed

• Bump jax from 0.4.36 to 0.4.37 in /mpi4jax/_src by @dependabot in #271

Full Changelog: v0.6.1.post1...v0.6.1.post2

What's Changed

• Bump jax from 0.4.35 to 0.4.36 in /mpi4jax/_src by @dependabot in #270

Full Changelog: v0.6.1...v0.6.1.post1

What's Changed

• Fix xpu backend by @dionhaefner in #266

Full Changelog: v0.6.0...v0.6.1