pymc.model.core.Model.profile#

Model.profile(outs, *, n=1000, point=None, profile=True, backend=None, **compile_fn_kwargs)[source]#

Compile and profile a PyTensor function which returns outs and takes values of model vars as a dict as an argument.

Parameters:
outsPyTensor variable or iterable of PyTensor variables
nint, default 1000

Number of iterations to run

pointPoint

Point to pass to the function

profileTrue or ProfileStats
backendstr, optional

Which computational backend to use. Recommended to be one of “numba”, “c”, and “jax”.

compile_fn_kwargs

Compilation kwargs for pymc.model.core.Model.compile_fn()

Returns:
pytensor.compile.profiling.ProfileStats

Use .summary() to print stats.