Update: I have been using Metropolis for the full model (with 142 sets of parameters) for now, but as expected, I get very correlated samples (only a few hundred effective samples after sampling for 50k samples). Any advice with NUTS on this problem will be very appreciated, thanks in advance for the help 