What's Changed
- DOC: Minor wording changes to numpyro code comments by @star1327p in #2133
- chore: files formatted with
ruff-v0.15.0by @Qazalbash in #2136 - test: update test effected by the
optax==0.2.7release by @Qazalbash in #2137 - test: adjust assertion for
my_fn_callsbased on optax version intest/test_optimizers.pyby @Qazalbash in #2140 - build:
setup.py->pyproject.tomlby @Qazalbash in #2141 - chore: replace legacy
jax.random.PRNGKeywith modernjax.random.keyby @Qazalbash in #2134 - Add PSIS k-hat diagnostic for variational inference by @michaelellis003 in #2139
- doc: Fix a few typos in numpyro code comments by @star1327p in #2144
- try fixing downloads datasets by @juanitorduz in #2151
- chore: remove
dm-haikusupport fromnumpyro.contrib.moduleby @theorashid in #2150 - Add GPJax notebook by @thomaspinder in #2149
- GPJax Notebook is correctly display in docs by @juanitorduz in #2152
- expose progress bar print rate option in MCMC by @mdmould in #2145
- Always call
jnp.finfoon arraydtypes, not arrays themselves by @dylanhmorris in #2147 - fix test because new jax 0.9.2 release by @juanitorduz in #2155
- CI(gh-2126): transition to uv for dependency management and testing by @Qazalbash in #2129
- fix test because new jax 0.9.2 release [v2] by @juanitorduz in #2156
- Use PyThon 3.14 in the Test Suite (CI) by @juanitorduz in #2157
- fix: remove frozen project build during doc generation by @Qazalbash in #2160
- Bump NumPyro 0.20.1 by @juanitorduz in #2159
New Contributors
- @michaelellis003 made their first contribution in #2139
- @thomaspinder made their first contribution in #2149
- @mdmould made their first contribution in #2145
Full Changelog: 0.20.0...0.20.1