Skip to content

Make nutpie the default nuts sample#8248

Draft
ricardoV94 wants to merge 1 commit intopymc-devs:v6from
ricardoV94:nutpie_default
Draft

Make nutpie the default nuts sample#8248
ricardoV94 wants to merge 1 commit intopymc-devs:v6from
ricardoV94:nutpie_default

Conversation

@ricardoV94
Copy link
Copy Markdown
Member

@ricardoV94 ricardoV94 commented Apr 12, 2026

Closes #8079
Closes #8111

First need to update nutpie to use modern pytensor imports, and more importantly arviz 1.0: pymc-devs/nutpie#297

@ricardoV94 ricardoV94 changed the title Make nutpies the default nuts sample Make nutpie the default nuts sample Apr 12, 2026
@ricardoV94 ricardoV94 added this to the v6 milestone Apr 12, 2026
@read-the-docs-community
Copy link
Copy Markdown

Documentation build overview

📚 pymc | 🛠️ Build #32224588 | 📁 Comparing c356b69 against latest (4b0b6ee)

  🔍 Preview build  

Show files changed (215 files in total): 📝 211 modified | ➕ 4 added | ➖ 0 deleted
File Status
genindex.html 📝 modified
glossary.html 📝 modified
api/backends.html 📝 modified
api/logprob.html 📝 modified
contributing/developer_guide.html 📝 modified
_modules/pymc/data.html 📝 modified
_modules/pymc/pytensorf.html 📝 modified
_modules/pymc/testing.html 📝 modified
api/distributions/multivariate.html 📝 modified
api/generated/pymc.backends.zarr.ZarrTrace.html 📝 modified
api/generated/pymc.compute_deterministics.html 📝 modified
api/generated/pymc.compute_log_likelihood.html 📝 modified
api/generated/pymc.compute_log_prior.html 📝 modified
api/generated/pymc.draw.html 📝 modified
api/generated/pymc.icdf.html 📝 modified
api/generated/pymc.logcdf.html 📝 modified
api/generated/pymc.logp.html 📝 modified
api/generated/pymc.ode.DifferentialEquation.html 📝 modified
api/generated/pymc.predictions_to_inference_data.html 📝 modified
api/generated/pymc.sample.html 📝 modified
api/generated/pymc.sample_posterior_predictive.html 📝 modified
api/generated/pymc.sample_prior_predictive.html 📝 modified
api/generated/pymc.sampling.jax.sample_blackjax_nuts.html 📝 modified
api/generated/pymc.sampling.jax.sample_numpyro_nuts.html 📝 modified
api/generated/pymc.smc.sample_smc.html 📝 modified
api/generated/pymc.to_inference_data.html 📝 modified
learn/core_notebooks/dimensionality.html 📝 modified
learn/core_notebooks/dims_module.html 📝 modified
learn/core_notebooks/model_comparison.html 📝 modified
learn/core_notebooks/posterior_predictive.html 📝 modified
learn/core_notebooks/pymc_overview.html 📝 modified
_modules/pymc/backends/arviz.html 📝 modified
_modules/pymc/backends/base.html 📝 modified
_modules/pymc/backends/ndarray.html 📝 modified
_modules/pymc/backends/zarr.html 📝 modified
_modules/pymc/distributions/continuous.html 📝 modified
_modules/pymc/distributions/custom.html 📝 modified
_modules/pymc/distributions/discrete.html 📝 modified
_modules/pymc/distributions/distribution.html 📝 modified
_modules/pymc/distributions/mixture.html 📝 modified
_modules/pymc/distributions/multivariate.html 📝 modified
_modules/pymc/distributions/shape_utils.html 📝 modified
_modules/pymc/distributions/timeseries.html 📝 modified
_modules/pymc/distributions/transforms.html 📝 modified
_modules/pymc/distributions/truncated.html 📝 modified
_modules/pymc/logprob/basic.html 📝 modified
_modules/pymc/model/core.html 📝 modified
_modules/pymc/model/fgraph.html 📝 modified
_modules/pymc/ode/ode.html 📝 modified
_modules/pymc/sampling/deterministic.html 📝 modified
_modules/pymc/sampling/forward.html 📝 modified
_modules/pymc/sampling/mcmc.html 📝 modified
_modules/pymc/smc/sampling.html 📝 modified
_modules/pymc/stats/log_density.html 📝 modified
_modules/pymc/tuning/starting.html 📝 modified
_modules/pymc/variational/approximations.html 📝 modified
_modules/pymc/variational/operators.html 📝 modified
_modules/pymc/variational/opvi.html 📝 modified
_modules/pytensor/tensor/basic.html 📝 modified
_modules/pytensor/tensor/extra_ops.html 📝 modified
_modules/pytensor/tensor/math.html 📝 modified
_modules/pytensor/tensor/special.html 📝 modified
api/dims/generated/pymc.dims.Beta.html 📝 modified
api/dims/generated/pymc.dims.Cauchy.html 📝 modified
api/dims/generated/pymc.dims.Gamma.html 📝 modified
api/dims/generated/pymc.dims.HalfCauchy.html 📝 modified
api/dims/generated/pymc.dims.HalfNormal.html 📝 modified
api/dims/generated/pymc.dims.HalfStudentT.html 📝 modified
api/dims/generated/pymc.dims.InverseGamma.html 📝 modified
api/dims/generated/pymc.dims.Laplace.html 📝 modified
api/dims/generated/pymc.dims.LogNormal.html 📝 modified
api/dims/generated/pymc.dims.Normal.html 📝 modified
api/dims/generated/pymc.dims.StudentT.html 📝 modified
api/dims/generated/pymc.dims.TruncatedNormal.html 📝 modified
api/dims/generated/pymc.dims.Uniform.html 📝 modified
api/dims/generated/pymc.dims.Weibull.html 📝 modified
api/distributions/classmethods/pymc.Censored.dist.html 📝 modified
api/distributions/classmethods/pymc.Simulator.dist.html 📝 modified
api/distributions/classmethods/pymc.Truncated.dist.html 📝 modified
api/distributions/generated/pymc.Bernoulli.html 📝 modified
api/distributions/generated/pymc.Beta.html 📝 modified
api/distributions/generated/pymc.BetaBinomial.html 📝 modified
api/distributions/generated/pymc.Binomial.html 📝 modified
api/distributions/generated/pymc.Categorical.html 📝 modified
api/distributions/generated/pymc.Cauchy.html 📝 modified
api/distributions/generated/pymc.ChiSquared.html 📝 modified
api/distributions/generated/pymc.Continuous.html 📝 modified
api/distributions/generated/pymc.Discrete.html 📝 modified
api/distributions/generated/pymc.DiscreteUniform.html 📝 modified
api/distributions/generated/pymc.DiscreteWeibull.html 📝 modified
api/distributions/generated/pymc.Distribution.html 📝 modified
api/distributions/generated/pymc.ExGaussian.html 📝 modified
api/distributions/generated/pymc.Exponential.html 📝 modified
api/distributions/generated/pymc.Gamma.html 📝 modified
api/distributions/generated/pymc.Geometric.html 📝 modified
api/distributions/generated/pymc.Gumbel.html 📝 modified
api/distributions/generated/pymc.HalfCauchy.html 📝 modified
api/distributions/generated/pymc.HalfNormal.html 📝 modified
api/distributions/generated/pymc.HalfStudentT.html 📝 modified
api/distributions/generated/pymc.HyperGeometric.html 📝 modified
api/distributions/generated/pymc.Interpolated.html 📝 modified
api/distributions/generated/pymc.InverseGamma.html 📝 modified
api/distributions/generated/pymc.Kumaraswamy.html 📝 modified
api/distributions/generated/pymc.LKJCorr.html 📝 modified
api/distributions/generated/pymc.Laplace.html 📝 modified
api/distributions/generated/pymc.LogNormal.html 📝 modified
api/distributions/generated/pymc.Logistic.html 📝 modified
api/distributions/generated/pymc.LogitNormal.html 📝 modified
api/distributions/generated/pymc.Moyal.html 📝 modified
api/distributions/generated/pymc.NegativeBinomial.html 📝 modified
api/distributions/generated/pymc.Normal.html 📝 modified
api/distributions/generated/pymc.Pareto.html 📝 modified
api/distributions/generated/pymc.Poisson.html 📝 modified
api/distributions/generated/pymc.PolyaGamma.html 📝 modified
api/distributions/generated/pymc.Rice.html 📝 modified
api/distributions/generated/pymc.SkewNormal.html 📝 modified
api/distributions/generated/pymc.StudentT.html 📝 modified
api/distributions/generated/pymc.SymbolicRandomVariable.html 📝 modified
api/distributions/generated/pymc.Triangular.html 📝 modified
api/distributions/generated/pymc.TruncatedNormal.html 📝 modified
api/distributions/generated/pymc.Uniform.html 📝 modified
api/distributions/generated/pymc.VonMises.html 📝 modified
api/distributions/generated/pymc.Wald.html 📝 modified
api/distributions/generated/pymc.Weibull.html 📝 modified
api/distributions/generated/pymc.ZeroInflatedBinomial.html 📝 modified
api/distributions/generated/pymc.ZeroInflatedNegativeBinomial.html 📝 modified
api/distributions/generated/pymc.ZeroInflatedPoisson.html 📝 modified
api/generated/classmethods/pymc.backends.NDArray.point.html 📝 modified
api/generated/classmethods/pymc.backends.base.BaseTrace.point.html 📝 modified
api/generated/classmethods/pymc.backends.zarr.ZarrChain.point.html 📝 modified
api/generated/classmethods/pymc.backends.zarr.ZarrTrace.to_inferencedata.html 📝 modified
api/generated/classmethods/pymc.ode.DifferentialEquation.L_op.html 📝 modified
api/generated/classmethods/pymc.ode.DifferentialEquation.R_op.html 📝 modified
api/generated/classmethods/pymc.ode.DifferentialEquation.do_constant_folding.html 📝 modified
api/generated/classmethods/pymc.ode.DifferentialEquation.grad.html 📝 modified
api/generated/classmethods/pymc.ode.DifferentialEquation.prepare_node.html 📝 modified
api/generated/classmethods/pymc.ode.DifferentialEquation.pullback.html ➕ added
api/generated/classmethods/pymc.ode.DifferentialEquation.pushforward.html ➕ added
_modules/pymc/dims/distributions/censored.html 📝 modified
_modules/pymc/dims/distributions/scalar.html 📝 modified
_modules/pymc/dims/distributions/vector.html 📝 modified
_modules/pymc/model/transform/optimization.html 📝 modified
api/distributions/generated/classmethods/pymc.AR.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.AsymmetricLaplace.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Bernoulli.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Beta.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.BetaBinomial.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Binomial.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.CAR.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Categorical.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Cauchy.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Continuous.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.DiracDelta.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Dirichlet.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.DirichletMultinomial.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Discrete.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.DiscreteUniform.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.DiscreteWeibull.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Distribution.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.EulerMaruyama.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.ExGaussian.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Exponential.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Flat.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.GARCH11.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Gamma.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Geometric.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Gumbel.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.HalfCauchy.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.HalfFlat.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.HalfNormal.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.HalfStudentT.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.HyperGeometric.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.ICAR.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Interpolated.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.InverseGamma.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.KroneckerNormal.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Kumaraswamy.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.LKJCorr.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Laplace.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.LogNormal.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Logistic.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.MatrixNormal.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Mixture.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Moyal.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Multinomial.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.MvNormal.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.MvStudentT.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.NegativeBinomial.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Normal.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Pareto.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Poisson.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.PolyaGamma.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Rice.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.SkewNormal.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.SkewStudentT.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.StickBreakingWeights.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.StudentT.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.SymbolicRandomVariable.L_op.html 📝 modified
api/distributions/generated/classmethods/pymc.SymbolicRandomVariable.R_op.html 📝 modified
api/distributions/generated/classmethods/pymc.SymbolicRandomVariable.do_constant_folding.html 📝 modified
api/distributions/generated/classmethods/pymc.SymbolicRandomVariable.grad.html 📝 modified
api/distributions/generated/classmethods/pymc.SymbolicRandomVariable.prepare_node.html 📝 modified
api/distributions/generated/classmethods/pymc.SymbolicRandomVariable.pullback.html ➕ added
api/distributions/generated/classmethods/pymc.SymbolicRandomVariable.pushforward.html ➕ added
api/distributions/generated/classmethods/pymc.SymbolicRandomVariable.rebuild_rv.html 📝 modified
api/distributions/generated/classmethods/pymc.SymbolicRandomVariable.update.html 📝 modified
api/distributions/generated/classmethods/pymc.Triangular.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.TruncatedNormal.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Uniform.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.VonMises.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Wald.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Weibull.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.Wishart.dist.html 📝 modified
api/distributions/generated/classmethods/pymc.ZeroSumNormal.dist.html 📝 modified
api/model/generated/classmethods/pymc.model.core.Model.compile_fn.html 📝 modified

@codecov
Copy link
Copy Markdown

codecov bot commented Apr 12, 2026

Codecov Report

❌ Patch coverage is 46.66667% with 16 lines in your changes missing coverage. Please review.
⚠️ Please upload report for BASE (v6@acad966). Learn more about missing BASE report.

Files with missing lines Patch % Lines
pymc/sampling/mcmc.py 46.66% 16 Missing ⚠️
Additional details and impacted files

Impacted file tree graph

@@          Coverage Diff          @@
##             v6    #8248   +/-   ##
=====================================
  Coverage      ?   88.61%           
=====================================
  Files         ?      124           
  Lines         ?    19981           
  Branches      ?        0           
=====================================
  Hits          ?    17707           
  Misses        ?     2274           
  Partials      ?        0           
Files with missing lines Coverage Δ
pymc/sampling/mcmc.py 67.59% <46.66%> (ø)
🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

Comment on lines +90 to +95
try:
import nutpie

NUTPIE_INSTALLED = True
except ImportError:
NUTPIE_INSTALLED = False
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
try:
import nutpie
NUTPIE_INSTALLED = True
except ImportError:
NUTPIE_INSTALLED = False
import importlib
NUTPIE_INSTALLED = bool(importlib.util.find_spec("nutpie"))

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Use nutpie by default (*conditions apply)

2 participants