Hi,
Looking at Numpyro code (eg primitives.py sample) there is _PYRO_STACK
I wander if _PYRO_STACK is initialized somewhere and when this statement is active/inactive?
# if no active Messengers, draw a sample or return obs as expected:
if not _PYRO_STACK:
Thanks