Initialzing NUTS with covariance matrix as mass matrix

@junpenglao I don’t think that is correct. You still don’t know the correct order.

You can use the dict -> array mapping from model.logp_dlogp_function([vars]).dict_to_array. That is the order NUTS uses internally.

1 Like