TypeError in NUTS

Using your method, it runs, probs is always zero in the trace – it never explores.