Stable logsumexp softmax gradients without rewrites - #2313

Open

ricardoV94 wants to merge 3 commits into

pymc-devs:mainfrom

ricardoV94:stable_logsumexp_softmax_grads

Commits on Jul 30, 2026

Only exclude symbolic_op_recognition inside SymbolicOp inner graphs
ricardoV94
committed
Make logsumexp and logaddexp stable by construction
ricardoV94
committed
Recognize log_softmax and rewrite exp(log_softmax) -> softmax
ricardoV94
committed