pytensor
pytensor copied to clipboard
Argmax can fail in numba mode
Description
If aixs = None numba will try to iterate over None and error
(TODO: Add MWE)