pymc-devs / pymc-experimental

https://pymc-experimental.readthedocs.io
Other
77 stars 49 forks source link

Use vectorize in `finite_discrete_marginal_rv_logp` #337

Closed ricardoV94 closed 5 months ago

ricardoV94 commented 5 months ago

This should be more performant and also faster to compile due to smaller computational graphs.

I've been waiting for the vectorize functionality to mature, and I think it's good now