Mpi4jax

What's Changed
* maximum version mpi4py <4 by PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/250
* Bump jax from 0.4.30 to 0.4.31 in /mpi4jax/_src by dependabot in https://github.com/mpi4jax/mpi4jax/pull/251
* fixthedocs by dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/252


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

What's Changed

* maximum version mpi4py <4 by PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/250


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

What's Changed
* Rename gpu -> cuda where relevant by PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/237
* Bump jax from 0.4.28 to 0.4.29 in /mpi4jax/_src by dependabot in https://github.com/mpi4jax/mpi4jax/pull/239
* Improve build errors when --no-build-isolation by PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/240
* Cleanup Sendrecv implementation by PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/238
* Louder warning if mpi4py/Cython not installed in non isolated builds by PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/242
* Fixes for cuda pjrt plugin by PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/241
* Update install instructions to reflect the new jax cuda wheel policy by PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/246
* Bump jax from 0.4.29 to 0.4.30 in /mpi4jax/_src by dependabot in https://github.com/mpi4jax/mpi4jax/pull/247


**Full Changelog**: https://github.com/mpi4jax/mpi4jax/compare/v0.5.0...v0.5.1

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

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 https://github.com/mpi4jax/mpi4jax/pull/236


**Full Changelog**: https://github.com/mpi4jax/mpi4jax/compare/v0.4.1...v0.5

What's Changed
* Bump jax from 0.4.26 to 0.4.27 in /mpi4jax/_src by dependabot in https://github.com/mpi4jax/mpi4jax/pull/233
* Bump jax from 0.4.27 to 0.4.28 in /mpi4jax/_src by dependabot in https://github.com/mpi4jax/mpi4jax/pull/234


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