document sampling related functions as users should import them

Author: OriolAbril

docs/source/api/samplers.rst

@@ -4,34 +4,19 @@ Samplers
 This submodule contains functions for MCMC and forward sampling.
 
 
-.. currentmodule:: pymc.sampling.forward
+.. currentmodule:: pymc
 
 .. autosummary::
    :toctree: generated/
 
+   sample
    sample_prior_predictive
    sample_posterior_predictive
    draw
-
-.. currentmodule:: pymc.sampling.deterministic
    compute_deterministics
-
-
-.. currentmodule:: pymc.sampling.mcmc
-
-.. autosummary::
-   :toctree: generated/
-
-   sample
    init_nuts
-
-.. currentmodule:: pymc.sampling.jax
-
-.. autosummary::
-   :toctree: generated/
-
-   sample_blackjax_nuts
-   sample_numpyro_nuts
+   sampling.jax.sample_blackjax_nuts
+   sampling.jax.sample_numpyro_nuts
 
 
 Step methods