>>> import jax
>>> import numpyro


>>> def model():
...     with numpyro.plate("n", 2):
...         numpyro.sample("x", numpyro.distributions.Normal())

>>> guide = numpyro.infer.autoguide.AutoDelta(model)

>>> numpyro.handlers.seed(guide, 9)()
>>> guide.plates["n"]._indices
Array([0, 1], dtype=int32)

>>> jax.jit(lambda: numpyro.handlers.seed(guide, 9)())()
>>> guide.plates["n"]._indices
Traced<ShapedArray(int32[2])>with<DynamicJaxprTrace>