Are there any additional parameters that can be used to make NUTS sample more efficiently?

I have problem, where the posterior is expected to have a single peak and the model evaluation is slow (I’m estimating the eigenvalues of a large matrix). A relatively simple version of our code can be found here - Cell `In[10]`

in the notebook contains the `model()`

. The NUTS sampler takes a long time for every iteration step. So my question is that, can the sampler be tweaked for my specific problem to achieve faster convergence?