Bump jax from 0.3.25 to 0.4.1 (#346)

Bumps [jax](https://github.com/google/jax) from 0.3.25 to 0.4.1.
- [Release notes](https://github.com/google/jax/releases)
- [Changelog](https://github.com/google/jax/blob/main/CHANGELOG.md)
- [Commits](jax-ml/jax@jax-v0.3.25...jax-v0.4.1)

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

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

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

Author: dependabot[bot]

poetry.lock

@@ -53,7 +53,7 @@ python = ">=3.8,<3.11"
 cclib = "^1"
 scipy = "^1.9"
 
-jax = {version = "^0.3", optional = true}
+jax = {version = "^0.4", optional = true}
 jaxlib = {version = "^0.3", optional = true}
 rich = {version = "^12", optional = true}
 thermo = {version = "^0.2", optional = true}