Example of benchmarking scan backends against pure implementations in the respective backend#1632

Open

Dekermanjian wants to merge 11 commits into

pymc-devs:v2from

Dekermanjian:benchmark_backends

Commits on Oct 5, 2025

added trus_inputs to pytensor functions, added multiple pytensor backends, added block_until_ready to jax calls
Dekermanjian
committed
updated renderer for reviewnb
Dekermanjian
committed

Commits on Oct 6, 2025

Updated how jax functions are blocked and updated jax fibonacci algorithm to prevent constant folding
Dekermanjian
committed

Commits on Oct 7, 2025

Commits on Oct 8, 2025

updated pytensor elemwise to exclude rewrite that removes scan, updated fori_loop to explicitly tell it not to constant fold
Dekermanjian
committed

Commits on Oct 12, 2025

updated jax fibonacci to ensure no constant folding. ensured all pytensor dprint included scan in the graph
Dekermanjian
committed

Commits on Oct 17, 2025

updated pytensor scan methods to always include fixed iterations in n_step arg, updated jax elemntwise to use fori_loop
Dekermanjian
committed

Commits on Oct 26, 2025

moved alarm checks outside of loop for numba CUSUM
Dekermanjian
committed

Commits on Oct 27, 2025

updated fibonacci jax to use fori_loop and reduced N_STEPS
Dekermanjian
committed