About NUTS sampling and energy plot

There is no good answer to that question because you usually dont directly control the potential. Usually you should improve your model by trying different reparameterization, standardizing the scale of the predictors, and use more informative prior to cut out the part of the parameter space that might be problematic (avoid heavy tail prior, etc).