Pyro Primitives .. automodule:: numpyro.primitives param .. autofunction:: numpyro.primitives.param sample .. autofunction:: numpyro.primitives.sample module .. autofunction:: numpyro.primitives.module