But it is definitely mentioned in the docstrings of sample_numpyro_nuts: pymc/sampling_jax.py at 6804c96cf2039780e167328d7476dc92f043554f · pymc-devs/pymc · GitHub
Whether by accident or not I don’t know.
But it is definitely mentioned in the docstrings of sample_numpyro_nuts: pymc/sampling_jax.py at 6804c96cf2039780e167328d7476dc92f043554f · pymc-devs/pymc · GitHub
Whether by accident or not I don’t know.