How to use Inverse Gaussian distribution in pyro?

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