JAX 0.8 and PyMC v5 compatibility - #164

Closed

MilesCranmer wants to merge 21 commits into

exoplanet-dev:mainfrom

MilesCranmer:fix-jax

Commits on Nov 23, 2025

fix: jax 0.8 compatibility
MilesCranmer
committed
d6bf4e8
fix: get pymc working on v5
MilesCranmer
committed
314ba3e
deps: lower to python 3.10
MilesCranmer
committed
f13d3a4
fix: JIT compat of `_do_compute`
MilesCranmer
committed
5790e09
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed
b14435b
deps: require python 3.11 for jaxlib compat
MilesCranmer
committed
8b36a42
ci: skip bad nox arg
MilesCranmer
committed
461fb35
ci: fix nox target version
MilesCranmer
committed
9957276
deps: fix missing jaxlib install
MilesCranmer
committed
7d5d97d
ci: fix missing jax install for pymc tests
MilesCranmer
committed
272344d

Commits on Nov 29, 2025

fix: generator will no longer add lines of spaces

MilesCranmer
and
dfm
committed
341c2f5
fix: re-generated xla_ops without api.h

MilesCranmer
and
dfm
committed
ffaff43
refactor: clean up error handling

MilesCranmer
and
dfm
committed
f6fc718
fix: `Primitive` import

MilesCranmer
and
dfm
committed
2644c1f
build: update cmake per review

MilesCranmer
and
dfm
committed
4fa899d
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed
f527c78
refactor: put axis into template

MilesCranmer
and
dfm
committed
7679b85
fix: add back nrhs==1 branch
MilesCranmer
committed
faa9b95
style: remove extra padding
MilesCranmer
committed
63dd782
deps: remove upper bound
MilesCranmer
committed
09d3ac4

Commits on Nov 30, 2025

chore: re-run generator
MilesCranmer
committed
7d71090