Jesse Grabowski

### Description I would be nice for debugging/benchmarking to be allowed to do `function(inputs, outputs, mode='python')`. This would use the `PerformLinker()` with default rewrites (excluding things like `blas_opt` that introduces...

Add error message to `NonZero` Op telling users it cannot be jitted in JAX

7

### Description jax mode is missing a dispatch for the `NonZero` `Op`. There's a `jnp.nonzero`, so it should be easy to do.

Allow vectorization of graphs with`IfElse`

8

### Description As the title suggests, graphs with `IfElse` will fail at runtime if you try to vectorize them. The error raised reminds me of #1425 ```py import pytensor.tensor as...

`numba_funcify_Elemwise` can end up with `parent_node` in its kwargs when vectorizing `OpFromGraph`

1

### Description Trying to compile a graph vectorized with `vectorize_graph` that contains an `OpFromGraph` into numba mode results in an error: ```py X = pt.dmatrix("X", shape=(None, None)) X_batched = pt.tensor("X",...

`OpFromGraph` falls back to object mode when output shape graph contains `Blockwise`

3

### Description When compiling a vectorized graph with an `OpFromGraph` into numba mode, shape inference fails if the shape of the output depends on blockwise operations. Example: ```py X =...

Support for sparse jacobians

6

### Description I really want to be able to call `pt.jacobian` and get back a sparse matrix. For very large systems of equations, the jacobian is essentially never going to...

Implement `BandedDot` `Op`

26

## Description This PR adds a `BandedDot` `Op` that uses `gbmv` to do matrix-vector multiplication for the case that A is a banded matrix. In my testing, I found that...

Remove thread-unsafe multiprocessing code from `pathfinder.py`

2

The Pathfinder code has some multiprocessing options that are not correctly handled. This was causing the CI to hang indefinitely, which isn't good. Pathfinder tests were disabled in #597, pending...

The `test_histogram_approx_cont` was [failing](https://github.com/pymc-devs/pymc-extras/actions/runs/18881754949/job/53926356230) with the following error on Windows: ``` 2025-10-29T00:57:05.6287038Z FAILED tests/test_histogram_approximation.py::test_histogram_approx_cont[ndims=2-dask=True] - pymc.sampling.parallel.ParallelSamplingError: Chain 1 failed with: overflow encountered in dot ``` The test was disabled in...

The `PyMCStateSpace` class asks the user to fill out an (admittedly) huge number of properties so the model can build the coords/dims: - param_names - data_names - param_info - data_info...