Avoiding non-NUTS samplers when broadcasting a multiplication operation

Broadcasting should be fine, the error message would be more useful if it said which Op is raising that NotImplementedError

1 Like