Initialzing NUTS with covariance matrix as mass matrix

In your case you need to do model.logp_dlogp_function([alpha, sigma]).dict_to_array to get the correct mapping.