primitives.rst

Pyro Primitives

.. automodule:: numpyro.primitives

param

.. autofunction:: numpyro.primitives.param

sample

.. autofunction:: numpyro.primitives.sample

plate

.. autoclass:: numpyro.primitives.plate

plate_stack

.. autofunction:: numpyro.primitives.plate_stack

subsample

.. autofunction:: numpyro.primitives.subsample

deterministic

.. autofunction:: numpyro.primitives.deterministic

factor

.. autofunction:: numpyro.primitives.factor

module

.. autofunction:: numpyro.primitives.module

flax_module

.. autofunction:: numpyro.contrib.module.flax_module

haiku_module

.. autofunction:: numpyro.contrib.module.haiku_module

random_flax_module

.. autofunction:: numpyro.contrib.module.random_flax_module

random_haiku_module

.. autofunction:: numpyro.contrib.module.random_haiku_module

scan

.. autofunction:: numpyro.contrib.control_flow.scan