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
stderrcalculation intest/test_distributions::test_entropy_samplesto 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
Predictiveand new random keys by @juanitorduz in #2073 - Add
cdfmethod 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.transformsModule by @Qazalbash in #2066 - DOC: Correct a typo: lets -> let's by @star1327p in #2087
- chore: update
numpyro.util.set_platformforJAX_PLATFORM_NAMEandJAX_PLATFORMSby @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
unconditioneffect 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-commithook by @Qazalbash in #2108 - feat(gh-2110): setup for
astral-sh/tyby @Qazalbash in #2111 - Fix NumPy 2.4 deprecation warning when loading JSB Chorales dataset by @juanitorduz in #2113
- Refine
tyconfig 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.constraintsModule by @Qazalbash in #2085 - Add Types to
numpyro.distributions.distributionby @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
MixtureSameFamilyhave 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