Distributions — NumPyro documentation
num.pyro.ai