pytensor_utils#

PyTensor utility functions.

Functions

extract_response_distribution(pymc_model, ...)

Extract the response distribution graph, conditioned on posterior parameters.

merge_models(models, *[, prefixes, merge_on])

Merge multiple PyMC models into a single model.

validate_unique_value_vars(model)

Validate that a model has unique, non-null value var names and 1:1 mappings.

Classes

ModelSamplerEstimator(*[, tune, draws, ...])

Estimate computational characteristics of a PyMC model using JAX/NumPyro.