Softmax Regression with shared coefficients between classes

@chartl Thank you so much, that definitely fixed it!

Is there any theoretical reason why pm.Categorical should fail in this case?