Numpyro

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

What's Changed
* Add images to tutorial gallery by juanitorduz in https://github.com/pyro-ppl/numpyro/pull/1799
* Fix typo at publish workflow that triggers the upload many times by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1798
* Add further entropy implementations. by tillahoffmann in https://github.com/pyro-ppl/numpyro/pull/1800
* Support multidimensional HSGP (numpyro.contrib.hsgp) by brendancooley in https://github.com/pyro-ppl/numpyro/pull/1803
* fix(contrib.hsgp): fix incorrect dim arg to partial spectral density fns by brendancooley in https://github.com/pyro-ppl/numpyro/pull/1808
* fix(contrib.hsgp): convert matern spectral density from frequency domain by brendancooley in https://github.com/pyro-ppl/numpyro/pull/1811
* example: multidimensional HSGP by brendancooley in https://github.com/pyro-ppl/numpyro/pull/1801
* filter out tests waiting for next tfp release by juanitorduz in https://github.com/pyro-ppl/numpyro/pull/1817
* Convert explicit looping to bit twiddling for nuts u-turn calculations by andrewdipper in https://github.com/pyro-ppl/numpyro/pull/1818
* Make figure in AR2 example reproducible by damonbayer in https://github.com/pyro-ppl/numpyro/pull/1816
* Memory reduction fixes for MCMC sampler by andrewdipper in https://github.com/pyro-ppl/numpyro/pull/1802
* Less than eq constaint by Qazalbash in https://github.com/pyro-ppl/numpyro/pull/1822
* Support vector lengthscales for RBF and Matern kernels by samanklesaria in https://github.com/pyro-ppl/numpyro/pull/1819
* Update jax.tree_util.tree_map to jax.tree.map by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1821
* Fixes `random_flax_module` with `flax.linen.BatchNorm` by juanitorduz in https://github.com/pyro-ppl/numpyro/pull/1823
* Allow for more general chain_method in MCMC by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1825
* Support UnpackTransform.inv via pack_fn by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1824
* Allow to use NeuTra on models with plates by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1826
* Bump to 0.15.1 by fehiepsi in https://github.com/pyro-ppl/numpyro/pull/1830

New Contributors
* andrewdipper made their first contribution in https://github.com/pyro-ppl/numpyro/pull/1818
* damonbayer made their first contribution in https://github.com/pyro-ppl/numpyro/pull/1816
* samanklesaria made their first contribution in https://github.com/pyro-ppl/numpyro/pull/1819

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