what about e.g. ```lambda x:np.argmax(np.abs(x))``` ? ;-)

the point, as you pointed out, it is not about argmax but rather about all other functionality 

and it would eliminate not needed sign logic (just use np.argmax or np.argmin depending on the goal)