0.22.0

What's Changed

• Fix effect handlers SWIG reference by @jsakv in #2184
• Fix incorrect HPDI docstring by @ordabayevy in #2183
• fix(gh-2181): ensure consistent behavior of Poisson.log_prob by casting rate to floating-point type by @Qazalbash in #2182
• doc: detailed documentation of Beta and BetaProportion distribution by @Qazalbash in #2185
• doc(gh-2187): add math explanations for Normal, Cauchy, and Exponential distributions by @kimjune01 in #2188
• dev: pre-commit hook to standardize toml format by @Qazalbash in #2186
• fix: freshly formatted toml files by @Qazalbash in #2190
• doc: replace XXX with Note in the code comments by @star1327p in #2191
• Fix incorrect statement in GammaPoisson log_prob docstring by @dylanhmorris in #2194
• Expose NegativeBinomial from top-level distributions module by @dylanhmorris in #2196
• Expose ZeroInflatedLogits and ZeroInflatedProbs from top-level distributions module by @Qazalbash in #2197
• Fix EnsembleSampler second split to use complementary inactive chains by @amifalk in #2189
• doc(gh-2187): enhance documentation for Bernoulli and Binomial distributions by @Qazalbash in #2192
• Fix path Boston housing dataset by @juanitorduz in #2200
• doc(gh-2187): enhance documentation for Chi2, DiscreteUniform, and Categorical distributions by @Metbcy in #2199
• doc(gh-2187): docstrings for Gumbel, Laplace, Logistic by @Metbcy in #2202
• Hurdle Distributions by @juanitorduz in #2195
• Fix enable_validation by @renecotyfanboy in #2201
• Fix GaussianStateSpace.variance shape when batch comes from initial_value by @tillahoffmann in #2205
• Migrate from MyPy to Ty by @juanitorduz in #2207
• Add HaarTransform and DiscreteCosineTransform by @juanitorduz in #2208
• Better casting eps by @juanitorduz in #2209
• Improve integration with external samplers by @juanitorduz in #2203
• Rename var -> variance for truncated distributions and close support. by @tillahoffmann in #2210
• Improve type hints for the HSGP contrib module by @juanitorduz in #2211
• docs: document NUTS regularize_mass_matrix by @ahfoysal in #2214
• Pin ty and improve type hints by @juanitorduz in #2218
• Miscellaneous bugfixes by @esennesh in #2215
• Avoid unnecessary deepcopy in random_*_module and promote_batch_shape by @kyo219 in #2222
• Fix stale trace count in test_mcmc_parallel_chain and run it in CI by @esennesh in #2223
• chore: format README with recent ruff==0.16.0 by @Qazalbash in #2228
• doc(gh-2187): docstrings for Dirichlet and Geometric by @kir-7 in #2227
• doc: update README for header structure by @Qazalbash in #2231
• doc: fix stale instructions in docs/README by @Qazalbash in #2232
• feat: compare branch speed against master and report it on the PR by @Qazalbash in #2233
• ci: add PR triage workflow for auto review requests and labels by @Qazalbash in #2234
• ci: use pull_request_target for PR triage workflow by @Qazalbash in #2237
• chore: workflow to clean pr after merging by @Qazalbash in #2239
• Avoid creating NaNs in binary_cross_entropy_with_logits by @miguelbiron in #2241
• ci: grant issues:write to pr-cleanup workflow by @Qazalbash in #2242
• Validate num_samples in MCMC initialization by @haansn08 in #2236
• fix NaN mixture log_prob gradient at zero weights by @esennesh in #2224
• Fix provenance tracking for closed-over constants on JAX 0.11.1 by @tandede in #2245
• Speed up posterior diagnostics (summary 1.7x, effective_sample_size 2.1x) by @kyo219 in #2251
• Drop the masked logsumexp reduction in mixture log_prob by @kyo219 in #2250
• Fix generator being appended instead of a list in scan carry_shapes by @kyo219 in #2248
• Avoid per-step host sync in the SVI progress-bar loop by @kyo219 in #2247
• Add cat constraint and CatTransform by @aswanth-07 in #2246
• Simplify the inverse of SimplexToOrderedTransform by @haansn08 in #2253
• Add regression test for render_model with masked observations (#2254) by @Agi-Asi in #2258
• Support batched scale_tril in LowerCholeskyAffine by @castor639 in #2260
• Rename Predictive's exclude_deterministic to condition_deterministic by @Agi-Asi in #2259
• Ensure distribution moments/samples are jax arrays by @tillahoffmann in #2206
• fix(distributions): handle zero-mean negative binomial by @aswanth-07 in #2256
• doc: link citation markers on the Contributed Code page by @IvanDobrovolsky in #2261
• Fix implicit Optional on GraphicalKernel.local_kernel_fns by @VenishPaneliya in #2263
• doc: math notes for HalfCauchy, HalfNormal, and Uniform by @gabrielemidulla in #2262
• Add TruncatedGamma distributions by @deeb01 in #2257
• doc(gh-2187): docstrings for LogNormal and LogUniform by @IvanDobrovolsky in #2267
• doc(gh-2187): mathematical details for GaussianRandomWalk by @jabrailkhalil in #2266
• docs(gh-2187): add mathematical details for Pareto and Weibull by @Rokesh2008 in #2265
• doc(gh-2187): mathematical details for InverseGamma and Kumaraswamy by @Yi-111-a in #2269
• doc(gh-2187): mathematical details for VonMises by @Yi-111-a in #2270
• refactor: keep truncation bounds out of the static treedef for the power-law distributions by @deeb01 in #2272
• doc(gh-2187): mathematical details for StudentT by @Yi-111-a in #2268
• doc: link citation markers in the contrib API docstrings by @IvanDobrovolsky in #2264
• update censoring notebook by @juanitorduz in #2235
• Make Gibbs kernels composable by @juanitorduz in #2249
• fix(distributions): keep Categorical entropy finite when a category has zero probability by @Dev-next-gen in #2276
• Prepare version 0.22 by @juanitorduz in #2275

New Contributors

• @jsakv made their first contribution in #2184
• @kimjune01 made their first contribution in #2188
• @Metbcy made their first contribution in #2199
• @ahfoysal made their first contribution in #2214
• @kyo219 made their first contribution in #2222
• @kir-7 made their first contribution in #2227
• @miguelbiron made their first contribution in #2241
• @haansn08 made their first contribution in #2236
• @tandede made their first contribution in #2245
• @aswanth-07 made their first contribution in #2246
• @Agi-Asi made their first contribution in #2258
• @castor639 made their first contribution in #2260
• @IvanDobrovolsky made their first contribution in #2261
• @VenishPaneliya made their first contribution in #2263
• @gabrielemidulla made their first contribution in #2262
• @deeb01 made their first contribution in #2257
• @jabrailkhalil made their first contribution in #2266
• @Rokesh2008 made their first contribution in #2265
• @Yi-111-a made their first contribution in #2269
• @Dev-next-gen made their first contribution in #2276

Full Changelog: 0.21.0...0.22.0