0.20.0

What's Changed

• numpyro.contrib.module: only collect mutables in nnx_module() when th… by @esennesh in #2061
• Use functional interface funsor.adjoint.adjoint to avoid tracer leak by @fehiepsi in #2002
• fix(tests): fix stderr calculation in test/test_distributions::test_entropy_samples to handle zero standard deviation by @Qazalbash in #2069
• Ensure compatibility NNX 0.11.2 by @juanitorduz in #2067
• Fix test for new NNX 0.12.0 by @juanitorduz in #2076
• Fix compatibility Predictive and new random keys by @juanitorduz in #2073
• Add cdf method for truncated distributions by @juanitorduz in #2074
• Fix get model relations when a param is a lambda function by @juanitorduz in #2078
• DOC: Correct typos in Hierarchical Forecasting by @star1327p in #2082
• DOC: Update the URL to Stochastic Volatility PyMC example by @star1327p in #2083
• DOC: Update MC Stan links in Examples by @star1327p in #2084
• Implement LeftCensoredDistribution and RightCensoredDistribution by @vanAmsterdam in #2081
• fix(gh-2036): MyPy Errors in numpyro.distributions.transforms Module by @Qazalbash in #2066
• DOC: Correct a typo: lets -> let's by @star1327p in #2087
• chore: update numpyro.util.set_platform for JAX_PLATFORM_NAME and JAX_PLATFORMS by @Qazalbash in #2091
• Implement IntervalCensoredDistribution by @vanAmsterdam in #2090
• DOC: Correct a few grammatical errors by @star1327p in #2097
• feat: add support for CUDA 12 and CUDA 13 by @Qazalbash in #2094
• Fix Beta with concentration1=1 gives nan log_prob at value=0 by @juanitorduz in #2089
• Add rational quadratic kernel to HSGP approximation by @juanitorduz in #2100
• Add uncondition effect handler by @juanitorduz in #2101
• Add Beta Negative Binomial distribution by @juanitorduz in #2102
• Add Inverse Wishart Distribution by @juanitorduz in #2103
• Fix uniform distribution log_prob support by @juanitorduz in #2095
• Bugfix/mutables leaves by @esennesh in #2105
• distributions.mixtures._MixtureBase: mixtures lack reparameterized sa… by @esennesh in #2106
• resolve deprecated warning from pre-commit hook by @Qazalbash in #2108
• feat(gh-2110): setup for astral-sh/ty by @Qazalbash in #2111
• Fix NumPy 2.4 deprecation warning when loading JSB Chorales dataset by @juanitorduz in #2113
• Refine ty config by @juanitorduz in #2114
• ci: pre-commit hook to auto-format yaml files by @Qazalbash in #2112
• doc(gh-2107): Improved documentation for Negative Binomial distributions by @Qazalbash in #2109
• Disable masking when running Predictive with infer_discrete=True by @fehiepsi in #2115
• DOC: Correct a few a/an usages by @star1327p in #2116
• FIX: JAX pxla is deprecated from 0.8.2 onwards by @juanitorduz in #2119
• Bump minimum python version to 3.11 and minimum jax version to 0.7 by @fehiepsi in #2120
• Add initial value to Gaussian state space distribution (fixes #2098). by @tillahoffmann in #2104
• fix(gh-2036): MyPy Errors in numpyro.distributions.constraints Module by @Qazalbash in #2085
• Add Types to numpyro.distributions.distribution by @juanitorduz in #2122
• Remove Typing Protocols by @juanitorduz in #2125
• ci: prek for pre-commit hooks check by @Qazalbash in #2121
• fix(gh-2123): Validate component distributions of MixtureSameFamily have parameter-independent support by @Qazalbash in #2127
• Bump numpyro to 0.20.0 by @fehiepsi in #2128
• Fix docs build for numpyro 0.20 by @fehiepsi in #2131

New Contributors

• @esennesh made their first contribution in #2061
• @star1327p made their first contribution in #2082

Full Changelog: 0.19.0...0.20.0