Bump jax from 0.6.2 to 0.7.0 in /mpi4jax/_src (#283)

* Bump jax from 0.6.2 to 0.7.0 in /mpi4jax/_src

Bumps [jax](https://github.com/jax-ml/jax) from 0.6.2 to 0.7.0.
- [Release notes](https://github.com/jax-ml/jax/releases)
- [Changelog](https://github.com/jax-ml/jax/blob/main/CHANGELOG.md)
- [Commits](jax-ml/jax@jax-v0.6.2...jax-v0.7.0)

---
updated-dependencies:
- dependency-name: jax
  dependency-version: 0.7.0
  dependency-type: direct:production
  update-type: version-update:semver-minor
...

Signed-off-by: dependabot[bot] <support@github.com>

* instantiate zeros

* bump python version for coverage test

---------

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Co-authored-by: Dion Häfner <mail@dionhaefner.de>

Remove token mode (#279)

* remove token mode

* fix test

* fix test

* I said fix

* raise if a token argument is passed

* update actions

* 🐛

Bump jax from 0.5.2 to 0.6.2 in /mpi4jax/_src (#282)

Fix default no token mode for older JAX versions (#278)

* Fix default no token mode for older JAX versions

If JAX < 0.5.1, the default no token mode does not work.
See issue #274 and:
  - #274
  - jax-ml/jax#26087

This commit sets that if JAX < 0.5.1, it switches to token mode.

* Downgrade jax version limit for no-token mode

* move linsolve test to separate module and use xfail

* update comment

---------

Co-authored-by: Dion Häfner <dion.haefner@simulation.science>

Bump jax from 0.5.1 to 0.5.2 in /mpi4jax/_src (#277)

Bumps [jax](https://github.com/jax-ml/jax) from 0.5.1 to 0.5.2.
- [Release notes](https://github.com/jax-ml/jax/releases)
- [Changelog](https://github.com/jax-ml/jax/blob/jax-v0.5.2/CHANGELOG.md)
- [Commits](jax-ml/jax@jax-v0.5.1...jax-v0.5.2)

---
updated-dependencies:
- dependency-name: jax
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

master is now main

Bump jax from 0.4.37 to 0.4.38 in /mpi4jax/_src (#272)

Bumps [jax](https://github.com/jax-ml/jax) from 0.4.37 to 0.4.38.
- [Release notes](https://github.com/jax-ml/jax/releases)
- [Changelog](https://github.com/jax-ml/jax/blob/main/CHANGELOG.md)
- [Commits](jax-ml/jax@jax-v0.4.37...jax-v0.4.38)

---
updated-dependencies:
- dependency-name: jax
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

Bump jax from 0.4.36 to 0.4.37 in /mpi4jax/_src (#271)

Bumps [jax](https://github.com/jax-ml/jax) from 0.4.36 to 0.4.37.
- [Release notes](https://github.com/jax-ml/jax/releases)
- [Changelog](https://github.com/jax-ml/jax/blob/main/CHANGELOG.md)
- [Commits](jax-ml/jax@jax-v0.4.36...jax-v0.4.37)

---
updated-dependencies:
- dependency-name: jax
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

Bump jax from 0.4.35 to 0.4.36 in /mpi4jax/_src (#270)

Bumps [jax](https://github.com/jax-ml/jax) from 0.4.35 to 0.4.36.
- [Release notes](https://github.com/jax-ml/jax/releases)
- [Changelog](https://github.com/jax-ml/jax/blob/main/CHANGELOG.md)
- [Commits](jax-ml/jax@jax-v0.4.35...jax-v0.4.36)

---
updated-dependencies:
- dependency-name: jax
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

xpu is back on the menu boys (#266)