Mpi4jax

What's Changed
* Fix xpu backend by dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/266


**Full Changelog**: https://github.com/mpi4jax/mpi4jax/compare/v0.6.0...v0.6.1

What's Changed
* Bump jax from 0.4.33 to 0.4.35 in /mpi4jax/_src by dependabot in https://github.com/mpi4jax/mpi4jax/pull/264
* Removed support for intel XPU (for the moment)
* Switch to jax FFI interface


**Full Changelog**: https://github.com/mpi4jax/mpi4jax/compare/v0.5.4...v0.6.0

What's Changed
* Bump jax from 0.4.31 to 0.4.32 in /mpi4jax/_src by dependabot in https://github.com/mpi4jax/mpi4jax/pull/255
* Update _latest_jax_version.txt to jax 0.4.33 by PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/258
* Fix docs by dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/259
* Ensure out_types is an interable on gpu by dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/261


**Full Changelog**: https://github.com/mpi4jax/mpi4jax/compare/v0.5.3...v0.5.4

What's Changed
* Update _latest_jax_version.txt to jax 0.4.33 by PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/258
* Fix docs by dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/259


**Full Changelog**: https://github.com/mpi4jax/mpi4jax/compare/v0.5.3.post1...v0.5.3.post2

What's Changed
* Bump jax from 0.4.31 to 0.4.32 in /mpi4jax/_src by dependabot in https://github.com/mpi4jax/mpi4jax/pull/255


**Full Changelog**: https://github.com/mpi4jax/mpi4jax/compare/v0.5.3...v0.5.3.post1

What's Changed
* Fix for mpi4py>=4 by dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/253


**Full Changelog**: https://github.com/mpi4jax/mpi4jax/compare/v0.5.2...v0.5.3