NUTS behavior on hierarchical.jl still weird