-
Problem: when running an analysis via Numpyro, an inordinate amount of time is spent tracing every function attached to the Model object, but most of these functions are not needed at sample-time
S…
-
**Describe the bug**
Arviz's `from_numpyro` function works on many model objects in numpyro, but not the officially supported bridge with JaxNS — `numpyro.contrib.nested_sampling` — because `Nested…
-
### Feature
Switch distribution libraries...again!
The problem with `tfp` is that if we want to run non-shared parameters we need `vmap` the apply function (over params and observations), but this…
-
It seems possible to convert tensor flow probability models to NumyPyro models, so would be nice to have a notebook demonstrating this https://num.pyro.ai/en/0.7.0/_modules/numpyro/contrib/tfp/distrib…
-
Greetings!
I've ported a subset of emcee functionality to the [NumPyro](https://github.com/pyro-ppl/numpyro) project under the sampler name [AIES](https://num.pyro.ai/en/latest/mcmc.html#numpyro.i…
-
in my research domain (healthcare) many outcomes are right-censored (aka survival analysis / time-to-event); the data for an individual patient are basically a tuple of (observed_time, event / censori…
-
`arviz.from_numpyro` has a group called `predictions` which is used for "Out of sample predictions." We should investigate whether and how to use this group.
https://python.arviz.org/en/stable/api/…
-
Hello,
I have completed a successful fit of 10 supernova images and am now trying to use the scene.sample() function to get a measurement for the flux of the supernova. I was basing my code off of …
-
### Description
Encountered a XLARuntimeError while running a basic numpyro program using jax-metal. The issue arises when I try running the MCMC sampler.
```python
import numpyro
import nump…
-
Hi, I get an error when I `MCMC.run` after `warmup` with `AIES`.
Here is the example
```
import jax
import jax.numpy as jnp
import numpyro
from numpyro.infer import MCMC, AIES
import numpyro.di…