We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 864ecb3 commit 765fdefCopy full SHA for 765fdef
pymc/sampling/jax.py
@@ -317,7 +317,6 @@ def sample_blackjax_nuts(
317
postprocessing_backend: Optional[str] = None,
318
postprocessing_chunks: Optional[int] = None,
319
idata_kwargs: Optional[Dict[str, Any]] = None,
320
- **kwargs,
321
) -> az.InferenceData:
322
"""
323
Draw samples from the posterior using the NUTS method from the ``blackjax`` library.
@@ -530,7 +529,6 @@ def sample_numpyro_nuts(
530
529
531
idata_kwargs: Optional[Dict] = None,
532
nuts_kwargs: Optional[Dict] = None,
533
534
535
536
Draw samples from the posterior using the NUTS method from the ``numpyro`` library.
0 commit comments