Can NUTS work with stochastic gradient / mini-batch?

Probably not… unless the mini-batches are sufficiently large.

Might need an extra friction term to damp the noise introduced by mini-batch.