Implement loglikelihood and gradient checkers for numpyro

[flo-schu]

This is a small addition for the numpyro backend, which implements checks that can be manually run to diagnose nans and infs in the log-likelihood function and the gradients.