Pytensor question using nnet.softmax()

[cluhmann]

Softmax should be in pm.math (so pm.math.softmax()). That is just an alias to the functionality in pytensor.tensor.special.softmax(). So you can use pt.special.softmax() if that’s easier.