Skip to content

Stable logsumexp softmax gradients without rewrites - #2313

Open
ricardoV94 wants to merge 3 commits into
pymc-devs:mainfrom
ricardoV94:stable_logsumexp_softmax_grads
Open

Stable logsumexp softmax gradients without rewrites#2313
ricardoV94 wants to merge 3 commits into
pymc-devs:mainfrom
ricardoV94:stable_logsumexp_softmax_grads

Recognize log_softmax and rewrite exp(log_softmax) -> softmax

0b9a773
Select commit
Loading
Failed to load commit list.
Sign in for the full log view
Check for changes
succeeded Jul 30, 2026 in 6s