Как я могу получить amax массива массивов numpy?

У меня есть массив массивов, и мне нужно получить amax для каждой строки.

[[-1.53286755e-02  2.93346513e-02 -7.66370669e-02 -2.19948404e-02]
 [-1.25247389e-02  1.43674174e-02 -6.83528483e-02 -1.07235834e-02]
 [-1.51151344e-02  3.38465981e-02 -7.61222318e-02 -1.83092393e-02]
 [ 4.46393713e-03  1.87037606e-02 -6.09790049e-02 -1.94273535e-02]
 [-1.64875668e-02  2.76340060e-02 -6.66617230e-02 -1.33732818e-02]
 [-3.99522949e-04  1.20752566e-02 -6.02767840e-02 -1.47377616e-02]
 [-6.86090253e-03  2.22653989e-02 -6.96686357e-02 -3.29462299e-03]
 [ 3.94986942e-03  3.54665779e-02 -7.09369779e-02 -6.11725636e-03]
 [ 3.49313393e-03  2.98086051e-02 -7.18421638e-02 -5.45137282e-03]
 [ 1.18993018e-02  2.50301771e-02 -6.93574101e-02  2.55537452e-04]
 [ 8.19707289e-04  6.50310218e-02 -1.46873474e-01  7.08404928e-03]
...

Мой массив выглядит так, и я хочу сделать с ним np.amax, чтобы получить максимальное значение для каждой строки.

Я пробовал best_q = np.amax(amax[0]), но это просто возвращает самое высокое значение во всем массиве.

Я хотел бы сделать это как-то векторизованным, чтобы сэкономить время обработки.