pymc.pytensorf.compile_pymc

pymc.pytensorf.compile_pymc(inputs, outputs, random_seed=None, mode=None, **kwargs)

Use pytensor.function with specialized pymc rewrites always enabled.

This function also ensures shared RandomState/Generator used by RandomVariables in the graph are updated across calls, to ensure independent draws.

Parameters:

inputs: list of TensorVariables, optional

Inputs of the compiled PyTensor function

outputs: list of TensorVariables, optional

Outputs of the compiled PyTensor function

random_seed: int, array-like of int or SeedSequence, optional

Seed used to override any RandomState/Generator shared variables in the graph. If not specified, the value of original shared variables will still be overwritten.

mode: optional

PyTensor mode used to compile the function