All pull requests

Pull requests

• fix(distributions): broadcast shared observations in Gaussian densities

  awaiting review
  #2413·antonsoo opened Nov 5, 2026Contributor·

  ·
  1

  2

• fix(distributions): infer RecursiveLinearTransform parameter batches

  awaiting review
  #2410·DiadkoShmek opened Nov 5, 2026·

  ·
  5

• Plate-aware discrete Gibbs updates in DiscreteHMCGibbs

  awaiting review
  #2408·bhaskargurram-ai opened Nov 4, 2026Contributor·

  ·
  1

  22

• fix(distributions): preserve ordered logistic tail masses

  awaiting review
  #2406·DiadkoShmek opened Nov 4, 2026·

  ·
  1

• fix(distributions): preserve GaussianStateSpace variance batch axes

  awaiting review
  #2405·DiadkoShmek opened Nov 4, 2026·

  ·
  8

• fix(distributions): align Cat constraint event masks

  awaiting review
  #2404·DiadkoShmek opened Nov 4, 2026·

  ·
  6

• fix: renyiELBO raises TypeError when model has no sample sites

  awaiting review
  #2403·alibb007 opened Nov 2, 2026·

  ·
  1

  3

• Keep bounded DirichletMultinomial counts in pytree data

  awaiting review
  #2401·DiadkoShmek opened Nov 2, 2026·

  ·
  11

• fix(infer): support vectorized SA chains across runs

  awaiting review
  #2393·tandede opened Oct 27, 2026Contributor·

  ·
  2

  1

• fix(distributions): make pytree flattening independent of the hash seed

  awaiting review
  #2389·ma9o opened Oct 25, 2026·

  ·
  16

• fix(distributions): return 0 and 1 from cdf outside the support

  awaiting review
  #2388·mahimn01 opened Oct 24, 2026·

  ·
  24

• fix(infer): group the latents of each plate element in AutoBatched guides

  awaiting review
  #2387·G26karthik opened Oct 23, 2026Contributor·

  ·
  5

• fix(distributions): stabilize zero-inflated probability likelihoods and gradients

  awaiting review
  #2385·antonsoo opened Oct 21, 2026Contributor·

  ·
  4

• fix(infer): accept per-site initialization callables

  awaiting review
  #2384·leninworld opened Oct 21, 2026Contributor·

  ·
  1

  11

• fix(distributions): bound MixtureGeneral component cdfs by their supports

  awaiting review
  #2383·mottopanikeiku opened Oct 20, 2026Contributor·

  ·
  9

• fix(contrib.hsgp): normalize multidimensional rational quadratic spectra

  awaiting review
  #2374·mottopanikeiku opened Oct 19, 2026Contributor·

  ·
  5

• fix: preserve zero-inflated support through enumeration

  awaiting review
  #2344·GreedyC opened Oct 13, 2026Contributor·

  ·
  1

  11

• fix(gh-2338): prevent repeated vmapping of sample function in HMC

  awaiting review
  #2339·ingolfuer opened Oct 10, 2026·

  ·
  1

  3

• fix(contrib/funsor): explain unmatched enumeration dimensions

  awaiting review
  #2285·deeb01 opened Sep 23, 2026Contributor·

  ·
  1

  6

• Add GammaGaussianHMM, LinearHMM, GaussianMRF and LinearHMMReparam

  awaiting reviewdiscussion
  #2274·juanitorduz opened Sep 17, 2026Collaborator·

  ·
  3

  

• Add numpyro.ops.gaussian and GaussianHMM with parallel Kalman marginalization

  awaiting reviewdiscussion
  #2273·juanitorduz opened Sep 17, 2026Collaborator·

  ·
  10

  

• Add SchechterMag distribution for astrophysical luminosity functions

  enhancement
  #2213·arham766 opened Jul 7, 2026·

  ·
  1

  15

• Implement log spectral density for squared exponential

  awaiting response
  #2143·elliottperryman opened Feb 26, 2026·

  ·
  5

• Microcanonical Langevin Monte Carlo

  awaiting response
  #2039·reubenharry opened Jun 23, 2025·

  ·
  4

• [add] Implementation of the Beta Negative Binomial Distribution

  awaiting response
  #2031·ShozenD opened May 20, 2025·

  ·
  2