I don’t think you can have a stochastic input to NUTS, how would it work?
Even ignoring the fact you can’t differentiate wrt to it, NUTS evaluates the logp graph multiple times per iteration. In every evaluation the value of the random node would change, and my guess is that would really screwup with NUTS?
CC @aseyboldt