Numpyro documentation

Pyro Primitives:

Indices and tables