doc: add nuts_kwargs to sample_numpyro_nuts docstring (#5764)

The `nuts_kwargs` argument is currently missing from the docstring of `sample_numpyro_nuts`. This just adds a simple statement to indicate that the keyword arguments are passed to `numpyro.infer.NUTS()`.

Author: jhrcook

pymc/sampling_jax.py

@@ -435,6 +435,8 @@ def sample_numpyro_nuts(
         Keyword arguments for :func:`arviz.from_dict`. It also accepts a boolean as value
         for the ``log_likelihood`` key to indicate that the pointwise log likelihood should
         not be included in the returned object.
+    nuts_kwargs: dict, optional
+        Keyword arguments for :func:`numpyro.infer.NUTS`.
 
     Returns
     -------