Limit number of function evaluations

pymc3 samplers allow to define the number of draws (=steps, it seems). However, some samplers perform multiple function evaluations per step, with the number of evaluations per step being auto-tuned https://docs.pymc.io/api/inference.html?highlight=nuts#pymc3.step_methods.hmc.nuts.NUTS.

For us, it would be good to be able to limit the number of function evaluations. Reason: We have computationally expensive ODE based models and work on shared infrastructure where we need to be able to plan rather precisely how long an inference will take / how many simulations we perform. Also a combination of wall time checking with saving the trace already while sampling might suffice.

Question: Is this possible?

1 Like