python version of the No-U-Turn Sampler (NUTS) from Hoffman & Gelman, 2011