Issues

• Truncated distributions: cdf is nan and two-sided log_prob is +inf when the truncation interval lies in the upper tail

  pyro-ppl/numpyro#2330 · Filter by author JoeyTan21 opened on Oct 7, 2026

• AIES re-evaluates the log density of the current walkers at every step

  bug
  pyro-ppl/numpyro#2301 · Filter by author gaetanoscandariato opened on Oct 1, 2026

• ZeroInflatedProbs excludes 0 from its support when the base support does not contain 0

  pyro-ppl/numpyro#2295 · Filter by author nicoloangileri opened on Sep 30, 2026

• Maybe something changed recently to cause CategoricalLogit to no longer validate -inf in argument?

  bug
  Bug

  pyro-ppl/numpyro#2282 · Filter by author jaberg opened on Sep 21, 2026
  3

• Port HMM from Pyro

  enhancement
  pyro-ppl/numpyro#2278 · Filter by author juanitorduz opened on Sep 19, 2026
  5

• Easier Use of Distributions Under vmap

  enhancement
  pyro-ppl/numpyro#2271 · Filter by author DanWaxman opened on Sep 14, 2026
  3

• Parallel enumeration: wrong posterior weights (or _pyro_dim KeyError) when site probs come from chained indexing T[a][b]

  pyro-ppl/numpyro#2252 · Filter by author ferjorosa opened on Aug 23, 2026
  3

• feature: Pydantic to validate configurations

  discussion
  pyro-ppl/numpyro#2238 · Filter by author Qazalbash opened on Aug 5, 2026
  1

• Expanding samplers included in numpyro

  enhancement
  pyro-ppl/numpyro#2226 · Filter by author hessammehr opened on Jul 21, 2026

• NUTS ~100–200× slower on numpyro 0.21.0 than 0.20.0 for a simple hierarchical Bernoulli model (GPU)

  bugperformance
  pyro-ppl/numpyro#2225 · Filter by author yingboli opened on Jul 21, 2026
  3

• Feature Request: Allow user to specify batch size in 'vectorization' chain method

  question
  pyro-ppl/numpyro#2204 · Filter by author ahuang314 opened on May 31, 2026
  7

• Enhance Distribution Documentation with Mathematical Explanations and Implementation Details

  documentationgood first issue
  Task

  pyro-ppl/numpyro#2187 · Filter by author Qazalbash opened on May 8, 2026
  3

• Some Out-Of-Sample predictions require manually removing certain sites

  question
  pyro-ppl/numpyro#2158 · Filter by author kylejcaron opened on Mar 21, 2026
  3

• Feature request: Stochastic Gradient-MCMC support (SGLD, SGHMC, etc.)

  enhancement
  pyro-ppl/numpyro#2075 · Filter by author mattlevine22 opened on Sep 26, 2025
  2

• Add WALNUTS (Within-Orbit Adaptive Leapfrog No-U-Turn Sampler) Implementation

  enhancement
  pyro-ppl/numpyro#2070 · Filter by author juanitorduz opened on Sep 14, 2025
  4

• Add Schechter distributions for astrophysical luminosity functions

  enhancement
  pyro-ppl/numpyro#2065 · Filter by author alserene opened on Aug 22, 2025
  7

• Allow setting initial values on autoguides with subsampling

  enhancement
  pyro-ppl/numpyro#2063 · Filter by author martp91 opened on Aug 18, 2025

• Jax throws error when tracing auto guide after passing it to get_model_relations

  bug
  pyro-ppl/numpyro#2062 · Filter by author mochar opened on Aug 14, 2025
  4

  

• support for custom support in TransformedDistribution

  enhancement
  pyro-ppl/numpyro#2043 · Filter by author Qazalbash opened on Jul 13, 2025
  6

• MyPy Errors in Distributions Module

  bug
  pyro-ppl/numpyro#2036 · Filter by author juanitorduz opened on Jun 7, 2025
  4

• Adding new samplers to NumPyro

  enhancement
  pyro-ppl/numpyro#2035 · Filter by author reubenharry opened on Jun 3, 2025
  6

• Check transformation domain in TransformedDistribution constructor?

  enhancement
  pyro-ppl/numpyro#1989 · Filter by author dylanhmorris opened on Feb 27, 2025
  2

• Performance enhancements for init_strategy may lead to unexpected behavior.

  enhancementrefactor
  pyro-ppl/numpyro#1970 · Filter by author tillahoffmann opened on Feb 4, 2025

• survival analysis / time-to-event analysis in numpyro

  enhancement
  pyro-ppl/numpyro#1909 · Filter by author vanAmsterdam opened on Nov 15, 2024
  12

• Flax BNN is several times slower in JAX 0.4.33 compared to JAX 0.4.31

  jaxperformance
  pyro-ppl/numpyro#1867 · Filter by author ziatdinovmax opened on Sep 25, 2024
  4