Hi, I need to use Inverse Gaussian distribution in my model, but I don’t see it in Pyro. I have checked the forum Inverse Gaussian distribution but I still don’t see it in numpyro.distributions. Can anyone show me how to use this distribution in numpyro? I need to do SVI with it.
from tensorflow_probability.substrates.jax import distributions as tfd
def model():
numpyro.sample("x", tfd.InverseGamma(loc=1.2, concentration=3.4))
1 Like