Update references and code for NUTS sampler.