What's Changed * Bump jax from 0.5.0 to 0.5.1 in /mpi4jax/_src by dependabot in https://github.com/mpi4jax/mpi4jax/pull/276 * Bump jax from 0.5.1 to 0.5.2 in /mpi4jax/_src by dependabot in https://github.com/mpi4jax/mpi4jax/pull/277
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](https://github.com/mpi4jax/mpi4jax/issues) any issues with these changes.