Release 0.20.1 · pyro-ppl/numpyro

What's Changed

DOC: Minor wording changes to numpyro code comments by @star1327p in #2133
chore: files formatted with ruff-v0.15.0 by @Qazalbash in #2136
test: update test effected by the optax==0.2.7 release by @Qazalbash in #2137
test: adjust assertion for my_fn_calls based on optax version in test/test_optimizers.py by @Qazalbash in #2140
build: setup.py -> pyproject.toml by @Qazalbash in #2141
chore: replace legacy jax.random.PRNGKey with modern jax.random.key by @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-haiku support from numpyro.contrib.module by @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.finfo on array dtypes, 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

Full Changelog: 0.20.0...0.20.1