Is there an equivalent to pyro’s sequential plate in numpyro? I see in numpyro plates are not iterable.
with numpyro.handlers.seed(rng_seed=0):
for i in numpyro.plate("p", 8):
numpyro.sample("x_{}".format(i), dist.Bernoulli(0.5))
Output:
---------------------------------------------------------------------------
TypeError Traceback (most recent call last)
/tmp/ipykernel_111183/893146523.py in <module>
1 with numpyro.handlers.seed(rng_seed=0):
----> 2 for i in numpyro.plate("p", 8):
3 x = numpyro.sample("x_{}".format(i), dist.Bernoulli(0.5))
4 print(x)
TypeError: 'plate' object is not iterable