@guvectorize(['void(float64[:], float64[:])'], '(n) -> ()', target=
    'parallel', nopython=True)
def row_sum_gu(input, output):
    output[0] = square_sum(input)

@jit(nopython=True)
def row_sum_jit(input_array, output_array):
    m, n = input_array.shape
for i in range(m):
    output_array[i] = square_sum(input_array[(i), :])
return output_array
numba - гувекторизация чуть быстрее, чем jit