Numpyro

What's Changed
* Remove cached `plates` from `AutoGuide`. by tillahoffmann in https://github.com/pyro-ppl/numpyro/pull/1976
* Fix typo and tweak wording in `get_dependencies()` docs by dylanhmorris in https://github.com/pyro-ppl/numpyro/pull/1975
* Navigate to our forum for discussions by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1978
* Unify `sample_posterior()` signatures by tare in https://github.com/pyro-ppl/numpyro/pull/1979
* Add guidelines for bug report and feature request by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1980
* Make seed handler stateless by default by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1983
* Update paths to data files by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1996
* Add `sum_sites` option to sum loss over sites or return as dict. by tillahoffmann in https://github.com/pyro-ppl/numpyro/pull/1995
* Improve rng default in the `contrib/model.py` module by juanitorduz in https://github.com/pyro-ppl/numpyro/pull/1992
* Fix padding calculation by tare in https://github.com/pyro-ppl/numpyro/pull/1993
* Apply `init_to_median` in unconstrained space. by tillahoffmann in https://github.com/pyro-ppl/numpyro/pull/1994
* Fix tracer leak in hmc by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1985
* Fix trace leakage in SL-DAIS by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1984
* Mark xfail for leaked tracer tests by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1997
* Fix tracer leak in svi with mutable state by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/2001
* Fix Memleak in ASVGD by OlaRonning in https://github.com/pyro-ppl/numpyro/pull/2003
* Flax NNX integration by juanitorduz in https://github.com/pyro-ppl/numpyro/pull/1990
* Pass value arg to optax, allowing use of reduce_on_plateau by zmbc in https://github.com/pyro-ppl/numpyro/pull/1974
* Add `CirculantNormal` distribution. by tillahoffmann in https://github.com/pyro-ppl/numpyro/pull/1988
* circulant gp notebook: minor improvements by juanitorduz in https://github.com/pyro-ppl/numpyro/pull/2004
* Fix behavior of `logdiffexp(jnp.inf, jnp.inf)`, add test that would have caught it. by dylanhmorris in https://github.com/pyro-ppl/numpyro/pull/2007
* Port hierarchical multivariate forecasting model from Pyro to NumPyro by juanitorduz in https://github.com/pyro-ppl/numpyro/pull/2006
* Bump to 0.18.0 by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/2009

New Contributors
* zmbc made their first contribution in https://github.com/pyro-ppl/numpyro/pull/1974

**Full Changelog**: https://github.com/pyro-ppl/numpyro/compare/0.17.0...0.18.0

What's Changed
* Only include `arg_constraints` in `pytree_data_fields` if they are not `lazy_property`s. by tillahoffmann in https://github.com/pyro-ppl/numpyro/pull/1929
* Ensure BlockNeuralAutoregressiveTransform maps real->real by danielward27 in https://github.com/pyro-ppl/numpyro/pull/1897
* Update coverall badge to point to the master branch by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1930
* Add openssf badge by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1931
* Add utility function to evaluate log density for individual sites. by tillahoffmann in https://github.com/pyro-ppl/numpyro/pull/1932
* Update link to jax repo by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1936
* Fix tracer leak in contrib.module by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1935
* Implement ICDF Methods for Truncated Distributions by TheSkyentist in https://github.com/pyro-ppl/numpyro/pull/1938
* Add type hints to the primitives module by juanitorduz in https://github.com/pyro-ppl/numpyro/pull/1940
* Add types types to handlers module by juanitorduz in https://github.com/pyro-ppl/numpyro/pull/1941
* Add types optim module by juanitorduz in https://github.com/pyro-ppl/numpyro/pull/1942
* Add batch and event shape to `repr`. by tillahoffmann in https://github.com/pyro-ppl/numpyro/pull/1946
* SteinVI: Recompute Score Function* For Each Particle Interaction in the Attractive Force by OlaRonning in https://github.com/pyro-ppl/numpyro/pull/1947
* Make `Delta.log_prob` jit-able on metal and raise explicit error for `Delta` sampling site during initialization. by tillahoffmann in https://github.com/pyro-ppl/numpyro/pull/1950
* Remove unnecessary device_put by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1952
* feat: Levy distribution by Qazalbash in https://github.com/pyro-ppl/numpyro/pull/1943
* SteinVI: New quadratic form kernel with Gauss Newton metric by OlaRonning in https://github.com/pyro-ppl/numpyro/pull/1953
* Make `pip install` calls compatible with `zsh` by dylanhmorris in https://github.com/pyro-ppl/numpyro/pull/1962
* Add `log1mexp` and `logdiffexp` functions by dylanhmorris in https://github.com/pyro-ppl/numpyro/pull/1960
* Fetch `rng_key` using `prng_key` message in block handler. by tillahoffmann in https://github.com/pyro-ppl/numpyro/pull/1957
* Add `ComplexTransform` and update `biject_to` registry. by tillahoffmann in https://github.com/pyro-ppl/numpyro/pull/1964
* Fix event dim of some arg constraints by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1967
* Addresses the deprecation messages of some jax.core functionalities by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1966
* Refactor test_gof into a separate file by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1965
* fix(tests): update tolerance levels and `PRNGKey` usage for improved test stability by Qazalbash in https://github.com/pyro-ppl/numpyro/pull/1959
* chore(ci): disable continue-on-error for all test jobs in CI workflow by Qazalbash in https://github.com/pyro-ppl/numpyro/pull/1968
* Do not create plates for observed sites in `AutoGuide`. by tillahoffmann in https://github.com/pyro-ppl/numpyro/pull/1972
* Fix promote_batch_shape logic to take batch shapes of all parameters by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1973
* Bump to 0.17.0 by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1969

New Contributors
* TheSkyentist made their first contribution in https://github.com/pyro-ppl/numpyro/pull/1938
* dylanhmorris made their first contribution in https://github.com/pyro-ppl/numpyro/pull/1962

**Full Changelog**: https://github.com/pyro-ppl/numpyro/compare/0.16.1...0.17.0

What's Changed
* Support Python 3.9 by juanitorduz in https://github.com/pyro-ppl/numpyro/pull/1923
* added note that sbvm is only accurate for conc <= 10k by OlaRonning in https://github.com/pyro-ppl/numpyro/pull/1924
* Clarify the composability of handlers with jax primities by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1926
* Add coveralls to CI by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1925
* Add Bayesian VAR(2) example script 1658 by aibit0111 in https://github.com/pyro-ppl/numpyro/pull/1915
* Bump to 0.16.1 by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1927

New Contributors
* aibit0111 made their first contribution in https://github.com/pyro-ppl/numpyro/pull/1915

**Full Changelog**: https://github.com/pyro-ppl/numpyro/compare/0.16.0...0.16.1

What's Changed
* Fix flaky test_change_point_x64 by GaetanLepage in https://github.com/pyro-ppl/numpyro/pull/1863
* Refactor `validate_args` method on distributions (fixes 1865). by tillahoffmann in https://github.com/pyro-ppl/numpyro/pull/1866
* gh-1870: Refactor `log_prob` method in `_MixtureBase` class to handle `-jnp.inf` by Qazalbash in https://github.com/pyro-ppl/numpyro/pull/1874
* use biased autocorrelation estimate by default (resolves 1785) by jonny-so in https://github.com/pyro-ppl/numpyro/pull/1877
* remove depreciated line from docs by juanitorduz in https://github.com/pyro-ppl/numpyro/pull/1882
* feat: add api compat arg to NS.get_samples by nstarman in https://github.com/pyro-ppl/numpyro/pull/1880
* Fixing nested sampling by renecotyfanboy in https://github.com/pyro-ppl/numpyro/pull/1871
* ci: fix pre-commit by nstarman in https://github.com/pyro-ppl/numpyro/pull/1892
* Bump CI to python 3.10 by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1890
* Refactoring SteinVI by OlaRonning in https://github.com/pyro-ppl/numpyro/pull/1883
* Fix handling of event dimensions in `ComposeTransform` (fixes 1893). by tillahoffmann in https://github.com/pyro-ppl/numpyro/pull/1894
* Correct event dimensions for `ReshapeTransform`. by tillahoffmann in https://github.com/pyro-ppl/numpyro/pull/1895
* Updated SteinVI docs by OlaRonning in https://github.com/pyro-ppl/numpyro/pull/1898
* Fixed norm const for SBVM by OlaRonning in https://github.com/pyro-ppl/numpyro/pull/1899
* CHANGED Stein BNN example to use prediction with noise (i.e. y_bnn -> y) by OlaRonning in https://github.com/pyro-ppl/numpyro/pull/1900
* [BUG] Latent exclude deterministic by juanitorduz in https://github.com/pyro-ppl/numpyro/pull/1901
* Allow callable for vectorization over particles on SVI by juanitorduz in https://github.com/pyro-ppl/numpyro/pull/1902
* Add Gaussian state space model distribution. by tillahoffmann in https://github.com/pyro-ppl/numpyro/pull/1904
* Proof of Concept: Types and MyPy by juanitorduz in https://github.com/pyro-ppl/numpyro/pull/1906
* Allow to collect nested dict keys in mcmc by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1905
* Add types for contrib.stochastic_support and improve the HSGP module by juanitorduz in https://github.com/pyro-ppl/numpyro/pull/1907
* [Types] Diagnostics and Utils by juanitorduz in https://github.com/pyro-ppl/numpyro/pull/1912
* Add method to retrieve distribution arguments. by tillahoffmann in https://github.com/pyro-ppl/numpyro/pull/1913
* Merge postprocess_fn into the fori_collect loop by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1910
* Hm/all deterministic by hessammehr in https://github.com/pyro-ppl/numpyro/pull/1914
* Bump to 0.16.0 by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1917

New Contributors
* GaetanLepage made their first contribution in https://github.com/pyro-ppl/numpyro/pull/1863
* jonny-so made their first contribution in https://github.com/pyro-ppl/numpyro/pull/1877
* nstarman made their first contribution in https://github.com/pyro-ppl/numpyro/pull/1880

**Full Changelog**: https://github.com/pyro-ppl/numpyro/compare/0.15.3...0.16.0

What's Changed
* Computing number of gradient evaluations example by andyElking in https://github.com/pyro-ppl/numpyro/pull/1839
* Enable pmap progress bar with cpu backend / remove deprecated host_callback by andrewdipper in https://github.com/pyro-ppl/numpyro/pull/1841
* NumPyro and Other Samplers Tutorial by juanitorduz in https://github.com/pyro-ppl/numpyro/pull/1842
* avoids setting jax tracer as lazy property attribute by sbidari in https://github.com/pyro-ppl/numpyro/pull/1843
* enable progress bar for multi-gpu by andrewdipper in https://github.com/pyro-ppl/numpyro/pull/1849
* Resolve numerical instability in entropy of `GeometricLogits`. by tillahoffmann in https://github.com/pyro-ppl/numpyro/pull/1852
* Generic `cdf` and `icdf` implementation for scalar `TransformedDistribution`s. by tillahoffmann in https://github.com/pyro-ppl/numpyro/pull/1853
* Raise error when using CircularReparam at observed site by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1856
* Update docs to make Predictive behavior more clear by tomwallis in https://github.com/pyro-ppl/numpyro/pull/1850
* Fix binominal distribution by InfinityMod in https://github.com/pyro-ppl/numpyro/pull/1860
* Fix the file path when rendering model by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1857
* Fix DiscreteUniform.enumerate_support with non-trivial batch shape by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1859
* Implementation of Doubly Truncated Power Law and Lower Truncated Power Law by Qazalbash in https://github.com/pyro-ppl/numpyro/pull/1807
* Bump to 0.15.3 by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1858

New Contributors
* andyElking made their first contribution in https://github.com/pyro-ppl/numpyro/pull/1839
* sbidari made their first contribution in https://github.com/pyro-ppl/numpyro/pull/1843
* tomwallis made their first contribution in https://github.com/pyro-ppl/numpyro/pull/1850
* InfinityMod made their first contribution in https://github.com/pyro-ppl/numpyro/pull/1860

**Full Changelog**: https://github.com/pyro-ppl/numpyro/compare/0.15.2...0.15.3

What's Changed
* Simplify the provenance logic to prepare for the removal of jax named shape by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1837
* Bump to 0.15.2 by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1840


**Full Changelog**: https://github.com/pyro-ppl/numpyro/compare/0.15.1...0.15.2