Numpyro returning error “Type error nuts object is not callable” when running mcmc

I am following the Gaussian Process example in Numpyro Website : https://num.pyro.ai/en/latest/examples/gp.html