v0.5.0

What's Changed

mpi4jax does not require the CUDA toolkit to be installed anymore. Instead, if the pypi Nvidia wheels are installed already (which is the new default behaviour of jax) we can link to them directly.

• Compile with pip Nvidia packages... by @PhilipVinc in #236

Full Changelog: v0.4.1...v0.5

v0.4.1.post1

What's Changed

• Bump jax from 0.4.26 to 0.4.27 in /mpi4jax/_src by @dependabot in #233
• Bump jax from 0.4.27 to 0.4.28 in /mpi4jax/_src by @dependabot in #234

Full Changelog: v0.4.1...v0.4.1.post1

v0.4.1

What's Changed

• Fix #229 by @dionhaefner in #230
• Extending mpi4jax with XPU support by @jczaja in #226

New Contributors

• @jczaja made their first contribution in #226

Full Changelog: v0.4.0.post6...v0.4.1

v0.4.0.post6

What's Changed

• Bump jax from 0.4.25 to 0.4.26 in /mpi4jax/_src by @dependabot in #231

Full Changelog: v0.4.0.post5...v0.4.0.post6

v0.4.0.post5

What's Changed

• Bump jax from 0.4.24 to 0.4.25 in /mpi4jax/_src by @dependabot in #228

Full Changelog: v0.4.0.post4...v0.4.0.post5

v0.4.0.post4

What's Changed

• Bump jax from 0.4.23 to 0.4.24 in /mpi4jax/_src by @dependabot in #227

Full Changelog: v0.4.0.post3...v0.4.0.post4

v0.4.0.post3

What's Changed

• Bump jax from 0.4.21 to 0.4.23 in /mpi4jax/_src by @dependabot in #224

Full Changelog: v0.4.0.post2...v0.4.0.post3

v0.4.0.post2

What's Changed

• Bump jax from 0.4.20 to 0.4.21 in /mpi4jax/_src by @dependabot in #223

Full Changelog: v0.4.0.post1...v0.4.0.post2

v0.4.0.post1

What's Changed

• bump flake8 version by @dionhaefner in #222
• Bump jax from 0.4.19 to 0.4.20 in /mpi4jax/_src by @dependabot in #221

Full Changelog: v0.4.0...v0.4.0.post1

v0.4.0

What's Changed

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 #212
• mpi4jax.experimental.auto_tokenize is removed.

Full Changelog: v0.3.15...0.4.0