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