Docstring for `numpyro.infer.NUTS` is missing the explanation of `regularize_mass_matrix`.
Docstring for
numpyro.infer.NUTSis missing the explanation ofregularize_mass_matrix.