Ensure PyBaMM safely handles optional JAX: validate jax/jaxlib versions and platform, warn instead of raising errors.#5398

Open

thegialeo wants to merge 19 commits intopybamm-team:mainfrom

thegialeo:issue-5381-jax-import-crash

Commits on Feb 26, 2026

Commits on Feb 27, 2026

isolate test_pybamm_import into a forked subprocess run
thegialeo
committed

Commits on Feb 28, 2026

Temporarily skip test_pybamm_import to investigate CI flakiness
thegialeo
committed

Commits on Mar 3, 2026

Commits on Mar 4, 2026

fix: normalize 4-part jax/jaxlib versions (e.g. 0.9.0.1) before range check in has_jax
thegialeo
committed

Commits on Mar 6, 2026

Merge branch 'main' into issue-5381-jax-import-crash
thegialeo
authored

Commits on Mar 11, 2026

Merge branch 'main' into issue-5381-jax-import-crash
thegialeo
authored