Updating softmax in rethinking_2 chapter 11#240

Open

DamonCharlesRoberts wants to merge 3 commits intopymc-devs:mainfrom

DamonCharlesRoberts:hotfix-issue-239

Commits on Jun 4, 2025

Switching to pymc.math.softmax implementation; adding axis=1 argument to the softmax function for rethinking_2 chapter 11 notebook.
DamonCharlesRoberts
committed

Commits on Jun 6, 2025