I am surprised that would work. I would imagine you need to hold the stochastic variable constant during the NUTS proposal. How will it use gradients if they change (discretely) at every evaluation?
The custom sampler would be just to hold the stochastic variable constant while NUTS updates.