pymc-devs/pymc
View on GitHubDOC: pm.sampling.jax functions are not included in PyMC docs
Open
#6,854 opened on Aug 11, 2023
docshelp wanted
Repository metrics
- Stars
- (7,926 stars)
- PR merge metrics
- (Avg merge 11d) (7 merged PRs in 30d)
Description
Issue with current documentation:
Various functions made available via pm.sampling.jax, including sample_numpyro_nuts() and sample_blackjax_nuts() do not appear in the documentation despite these functions having docstrings. Maybe we didn't include the pm.sampling.jax module is some auto-docing because it was a bit experimental at the time?