Definition of a stochastic function

numpyro
1

I have a stochastic function that returns a sample which I am calling via numpyro.sample. However I am getting complaints with respect to the function’s missing attributes like batch_sample and expand. Does this imply I need to wrap my stochastic function in Distribution? Docs for sample indicate:

fn – a stochastic function that returns a sample.

2

Yes I guess you can inherit this class numpyro.distributions.distribution — NumPyro documentation

1 Like