[DEPENDABOT]: Bump numpyro from 0.11.0 to 0.12.1

[dependabot]

Bumps numpyro from 0.11.0 to 0.12.1.

Release notes

Sourced from numpyro's releases.

0.12.1

This release includes a fix for jax 0.4.11 (#1595).

0.12.0

New Features

• New distribution: Gompertz distribution (#1551)
• New initialization strategy: init_to_mean (#1550)
• New examples and tutorials:
  • Illustrate the usage of JAX PositionalSharding for distributing the computations of log_prob/grad over multiple devices in MCMC (#1514)
  • A port of Gaussian Mixture Model tutorial from Pyro (#1562)
  • A toy mixture model with discrete enumeration (#1568)
• New inference utilities get_transforms and unconstrain_fn to transform between unconstrained and constrained space (#1564)
• Support jaxns>=2.0.1 (#1546)

Enhancements and Bug Fixes

• Make transforms jittable (#1575)
• Fixed typo in surrogate posterior of beta (#1591)
• Do not scale mnist label (#1589)
• Do not mutate shapes of ExpandedDistribution for map-free ops (#1574)
• Add support for JAX custom PRNG (#1587)
• Include deterministic variables in AutoDelta's sample_posterior (#1584)
• Fix forward shape of SimplexToOrderTransform (#1583)
• Fix inf's in TruncatedNormal log_prob & sample (#1581)
• Allow users to specify total_count_max in Multinomial (#1557)
• Allow pickled mcmc object to run post warmup phase (#1558)
• Add init_params argument to svi.init() and svi.run() (#1561)
• Support pickling MCMC objects with enumeration (#1577)
• Raise error when reparameterize lognormal (#1548)
• Avoid initializing model params when already specified in guide (#1553)
• Respect init params if provided to mcmc.run (#1547)
• Fix provenance for jax 0.4.4 (#1543)
• Use analytic kl divergence in TraceEnum_ELBO (#1533)
• Properly handle contraction of guide plates in TraceEnum_ELBO (#1537)
• Raise an error if there is no common scale when model enumerated (#1536)
• Optimize reduction of enumerated guide sites (#1531)
• Guess max_plate_nesting in TraceEnum_ELBO (#1528)

Commits

• 5291d06 Bump to version 0.12.1 (#1597)
• e715881 Fix for jax>=0.4.7 (#1595)
• f981b29 Bump to version 0.12.0 (#1593)
• eab63ed Jittable transforms (#1575)
• e230805 fixed typo in surrogate posterior of beta (#1591)
• d897d7a do not scale mnist label (#1589)
• db2913b Do not mutate shapes of ExpandedDistribution for map-free ops (#1574)
• a8004f0 Add support for JAX custom PRNG (#1587)
• c6fb104 Include deterministic variables in AutoDelta's sample_posterior (#1584)
• a2c28c1 add forward shape for SimplexToOrderTransform (#1583)
• Additional commits viewable in compare view

Dependabot will resolve any conflicts with this PR as long as you don't alter it yourself. You can also trigger a rebase manually by commenting @dependabot rebase.

---

Dependabot commands and options

You can trigger Dependabot actions by commenting on this PR:

• @dependabot rebase will rebase this PR
• @dependabot recreate will recreate this PR, overwriting any edits that have been made to it
• @dependabot merge will merge this PR after your CI passes on it
• @dependabot squash and merge will squash and merge this PR after your CI passes on it
• @dependabot cancel merge will cancel a previously requested merge and block automerging
• @dependabot reopen will reopen this PR if it is closed
• @dependabot close will close this PR and stop Dependabot recreating it. You can achieve the same result by closing it manually
• @dependabot ignore this major version will close this PR and stop Dependabot creating any more for this major version (unless you reopen the PR or upgrade to it yourself)
• @dependabot ignore this minor version will close this PR and stop Dependabot creating any more for this minor version (unless you reopen the PR or upgrade to it yourself)
• @dependabot ignore this dependency will close this PR and stop Dependabot creating any more for this dependency (unless you reopen the PR or upgrade to it yourself)