Apply reduction along specific axes

I guess you mean maximum, which does in fact accept an axis argument. max does something else.