Ricardo Vieira

### Description It's useful to know when `arange` or `eye` is in a graph, but if these have constant inputs they will be constant folded, and then we would need...

Allow explicit RNG and Sparse input types in JAX functions

1

I will sort out Shared Variables in a separate PR

Add Feature that can go back and forward in rewrite history

1

## Description I can imagine this being very powerful for debugging or reasoning about rewrites? ```python import pytensor import pytensor.tensor as pt from pytensor.graph.fg import FunctionGraph from pytensor.graph.features import FullHistory...

Small improvements to docs API

4

## Description Small changes, just to get a feel for it.

We have some other rewrites that will push Alloc below Elemwise, so that we don't compute on repeated inputs, but this won't happen if there's an expand_dims in the way....

### Description Reshape has only two inputs, x, and a vector of the output shape. This is cumbersome because many times we want to analyze the individual dimensions to rewrite...

### Description We could add a method to the Ops, similar to infer_shape that infers the number of flops and memory usage of an Op given its inputs and input...

### Description For discoverability. We can raise `NotImplementedError` by default. Or create a subclass of ShapedOp?

### Description Failed in https://github.com/pymc-devs/pytensor/actions/runs/9712864428/job/26808478897?pr=871#step:6:3845 May need a seed?

### Description We are defaulting to the numpyro implementation when that's installed, but the numpyro implementation is incredibly wasteful for large N, as it consists of doing n Categorical draws...