pyro-ppl/numpyro

View on GitHub

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

Stars: 2.8k
Forks: 315
Open beginner issues: 0
Indexed issues: 48
Avg merge: 3d 9h
Merged PRs (30d): 27

Dominant language: Python
License: Apache-2.0
Last GitHub push: Sep 18, 2026
Latest indexed: Sep 19, 2026
Contributing guide: Contributing guide
Code of conduct: No code of conduct
Beginner labels: help wanted good first issue

Beginner friendly Everything

Open All

48 open issues indexed

Port HMM from Pyro Open

enhancement

Difficulty 4/5 3-5 days Newbie friendliness 25/100

pyro-ppl/numpyro#2278 · Opened Sep 19, 2026
Easier Use of Distributions Under `vmap` Open

enhancement

Difficulty 5/5 Over a week Newbie friendliness 35/100

pyro-ppl/numpyro#2271 · 3 comments · 4 reactions · Opened Sep 14, 2026
Parallel enumeration: wrong posterior weights (or _pyro_dim KeyError) when site probs come from chained indexing T[a][b] Open

Difficulty 4/5 3-5 days Newbie friendliness 55/100

pyro-ppl/numpyro#2252 · 2 comments · Opened Aug 23, 2026
feature: Pydantic to validate configurations Open

discussion

Difficulty 5/5 Over a week Newbie friendliness 25/100

pyro-ppl/numpyro#2238 · 1 comment · Opened Aug 5, 2026
Expanding samplers included in numpyro Open

enhancement

Difficulty 5/5 Over a week Newbie friendliness 25/100

pyro-ppl/numpyro#2226 · Opened Jul 21, 2026
NUTS ~100–200× slower on numpyro 0.21.0 than 0.20.0 for a simple hierarchical Bernoulli model (GPU) Open

bug performance

Difficulty 4/5 3-5 days Newbie friendliness 48/100

pyro-ppl/numpyro#2225 · 3 comments · Opened Jul 21, 2026
Feature Request: Allow user to specify batch size in 'vectorization' chain method Open

question

Difficulty 3/5 1-2 days Newbie friendliness 65/100

pyro-ppl/numpyro#2204 · 7 comments · 1 reaction · Opened May 31, 2026
Enhance Distribution Documentation with Mathematical Explanations and Implementation Details Open

documentation good first issue

Difficulty 4/5 3-5 days Newbie friendliness 45/100

pyro-ppl/numpyro#2187 · 3 comments · 1 reaction · Opened May 8, 2026
Some Out-Of-Sample predictions require manually removing certain sites Open

question

Difficulty 4/5 3-5 days Newbie friendliness 45/100

pyro-ppl/numpyro#2158 · 3 comments · Opened Mar 21, 2026
Feature request: Stochastic Gradient-MCMC support (SGLD, SGHMC, etc.) Open

enhancement

Difficulty 5/5 Over a week Newbie friendliness 35/100

pyro-ppl/numpyro#2075 · 2 comments · Opened Sep 26, 2025
Add WALNUTS (Within-Orbit Adaptive Leapfrog No-U-Turn Sampler) Implementation Open

enhancement

Difficulty 5/5 Over a week Newbie friendliness 25/100

pyro-ppl/numpyro#2070 · 1 comment · 10 reactions · Opened Sep 14, 2025
Add Schechter distributions for astrophysical luminosity functions Open

enhancement

Difficulty 5/5 Over a week Newbie friendliness 48/100

pyro-ppl/numpyro#2065 · 7 comments · 1 reaction · Opened Aug 22, 2025
Allow setting initial values on autoguides with subsampling Open

enhancement

Difficulty 4/5 3-5 days Newbie friendliness 38/100

pyro-ppl/numpyro#2063 · Opened Aug 18, 2025
Jax throws error when tracing auto guide after passing it to get_model_relations Open

bug

pyro-ppl/numpyro#2062 · 4 comments · 1 assignee · Opened Aug 14, 2025
support for custom `support` in `TransformedDistribution` Open

enhancement

Difficulty 4/5 3-5 days Newbie friendliness 35/100

pyro-ppl/numpyro#2043 · 6 comments · Opened Jul 13, 2025
MyPy Errors in Distributions Module Open

bug

Difficulty 4/5 3-5 days Newbie friendliness 25/100

pyro-ppl/numpyro#2036 · 4 comments · 1 reaction · Opened Jun 7, 2025
Adding new samplers to NumPyro Open

enhancement

Difficulty 5/5 Over a week Newbie friendliness 30/100

pyro-ppl/numpyro#2035 · 6 comments · 2 reactions · Opened Jun 3, 2025
Check transformation domain in `TransformedDistribution` constructor? Open

enhancement

Difficulty 4/5 3-5 days Newbie friendliness 35/100

pyro-ppl/numpyro#1989 · 2 comments · Opened Feb 27, 2025
Performance enhancements for `init_strategy` may lead to unexpected behavior. Open

enhancement refactor

Difficulty 3/5 1-2 days Newbie friendliness 35/100

pyro-ppl/numpyro#1970 · Opened Feb 4, 2025
survival analysis / time-to-event analysis in numpyro Open

enhancement

Difficulty 5/5 Over a week Newbie friendliness 25/100

pyro-ppl/numpyro#1909 · 12 comments · Opened Nov 15, 2024
Flax BNN is several times slower in JAX 0.4.33 compared to JAX 0.4.31 Open

jax performance

Difficulty 4/5 3-5 days Newbie friendliness 35/100

pyro-ppl/numpyro#1867 · 4 comments · Opened Sep 25, 2024
Samples are outside the support for DiscreteUniform distribution Open

bug

Difficulty 4/5 3-5 days Newbie friendliness 35/100

pyro-ppl/numpyro#1834 · 3 comments · Opened Jul 22, 2024
Stress test utility for numpyro? Open

enhancement

Difficulty 5/5 Over a week Newbie friendliness 30/100

pyro-ppl/numpyro#1833 · 2 comments · 3 reactions · Opened Jul 16, 2024
mean_accept_prob significantly different after warmup Open

enhancement help wanted

Difficulty 4/5 3-5 days Newbie friendliness 38/100

pyro-ppl/numpyro#1786 · 10 comments · Opened Apr 19, 2024
Factor Analysis/PPCA Tutorial Open

Tutorials/Examples

Difficulty 4/5 3-5 days Newbie friendliness 38/100

pyro-ppl/numpyro#1768 · 1 comment · 2 reactions · Opened Mar 23, 2024
numpyro.render_model render unexpected graph with handler.do Open

discussion

Difficulty 4/5 3-5 days Newbie friendliness 35/100

pyro-ppl/numpyro#1716 · 5 comments · Opened Jan 9, 2024
GMM notebook example: MCMC/NUTS simulation is not reproducible Open

help wanted Tutorials/Examples

Difficulty 4/5 3-5 days Newbie friendliness 35/100

pyro-ppl/numpyro#1616 · 2 comments · Opened Jun 27, 2023
Multi-chain init to value Open

enhancement

Difficulty 5/5 Over a week Newbie friendliness 25/100

pyro-ppl/numpyro#1615 · 4 reactions · Opened Jun 27, 2023
init_strategy does not work with CircularReparam Open

enhancement

Difficulty 4/5 3-5 days Newbie friendliness 35/100

pyro-ppl/numpyro#1614 · 1 comment · Opened Jun 23, 2023
[FR] Render models to a LaTex formula Open

enhancement

Difficulty 5/5 Over a week Newbie friendliness 30/100

pyro-ppl/numpyro#1563 · 1 reaction · Opened Mar 26, 2023
`SkewMultivariateNormal` and `SkewMultivariateStudentT` Open

enhancement

Difficulty 5/5 Over a week Newbie friendliness 25/100

pyro-ppl/numpyro#1452 · 4 comments · Opened Jul 17, 2022
[FR] make DiscreteHMCGibbs plate aware Open

enhancement

Difficulty 4/5 3-5 days Newbie friendliness 35/100

pyro-ppl/numpyro#1406 · 1 comment · 1 reaction · Opened May 14, 2022
Non-bijective deterministic objective Open

discussion

Difficulty 5/5 Over a week Newbie friendliness 25/100

pyro-ppl/numpyro#1405 · 1 comment · Opened May 14, 2022
Amortizing overhead in `.run` Open

enhancement help wanted jax

Difficulty 4/5 3-5 days Newbie friendliness 25/100

pyro-ppl/numpyro#1249 · 1 comment · 1 reaction · Opened Dec 10, 2021
Request for dynamic `num_warmup` argument to MCMC object Open

enhancement

Difficulty 5/5 Over a week Newbie friendliness 25/100

pyro-ppl/numpyro#1211 · Opened Nov 4, 2021
Automatic structured variational inference Open

enhancement help wanted

Difficulty 5/5 Over a week Newbie friendliness 25/100

pyro-ppl/numpyro#1117 · 11 comments · 1 reaction · Opened Aug 5, 2021
Support multiple tensor backends via NEP-47 (Python array API standard) Open

discussion

Difficulty 5/5 Over a week Newbie friendliness 25/100

pyro-ppl/numpyro#1083 · 6 comments · Opened Jul 4, 2021
Make Gibbs kernels composable Open

enhancement help wanted

pyro-ppl/numpyro#898 · 2 comments · 1 assignee · Opened Jan 27, 2021
Using numpyro primitives in jax's control flows fori_loop, while_loop, and cond Open

enhancement

Difficulty 4/5 3-5 days Newbie friendliness 25/100

pyro-ppl/numpyro#879 · 4 comments · Opened Jan 16, 2021
Feature Request: Guidelines on implementing Sparse GP in Numpyro Open

good first issue Tutorials/Examples

Difficulty 5/5 Over a week Newbie friendliness 25/100

pyro-ppl/numpyro#829 · 3 comments · 2 reactions · Opened Nov 27, 2020
Support weighted statistics in `summary` diagnostics Open

enhancement good first issue

Difficulty 4/5 3-5 days Newbie friendliness 35/100

pyro-ppl/numpyro#810 · 6 comments · 1 reaction · Opened Nov 9, 2020
Bayesian Additive Regression Trees Open

Tutorials/Examples

Difficulty 5/5 Over a week Newbie friendliness 25/100

pyro-ppl/numpyro#793 · 1 comment · 2 reactions · Opened Oct 31, 2020
Vectorized interpretation for pyro.scan Open

enhancement

pyro-ppl/numpyro#686 · 3 comments · 2 reactions · 1 assignee · Opened Jul 15, 2020
Support for Empirical distributions Open

enhancement

Difficulty 5/5 Over a week Newbie friendliness 25/100

pyro-ppl/numpyro#685 · 4 comments · Opened Jul 15, 2020
Implement `RelaxedOneHotCategoricalStraightThrough` Open

enhancement

Difficulty 4/5 3-5 days Newbie friendliness 35/100

pyro-ppl/numpyro#559 · 3 comments · 1 reaction · Opened Mar 13, 2020
Tutorial or example on embarrassingly parallel/consensus MCMC Open

low priority Tutorials/Examples

pyro-ppl/numpyro#417 · 2 comments · 4 reactions · 1 assignee · Opened Oct 28, 2019
Use type hints whenever possible Open

low priority refactor

Difficulty 5/5 Over a week Newbie friendliness 20/100

pyro-ppl/numpyro#299 · 6 comments · Opened Aug 27, 2019
Call for user contributed examples and tutorials Open

good first issue Tutorials/Examples

Difficulty 5/5 Over a week Newbie friendliness 25/100

pyro-ppl/numpyro#189 · Opened Jun 4, 2019

pyro-ppl/numpyro

Get new issues in your inbox