Вот что я сейчас делаю, это работает, но немного громоздко:

x = np.matrix([[1, 1], [2, -3]])
xmax = x.flat[abs(x).argmax()]