import numba import numpy as np @numba.jit(nopython=True) def minmax(array): N = array.shape[0] vmin = np.full((N,), np.inf,dtype=array.dtype) vmax = np.full((N,),-np.inf,dtype=array.dtype) for ii in range(N): for val in array[ii,...].ravel(): vmin[ii] = min(vmin[ii],val) vmax[ii] = max(vmax[ii],val) return vmin,vmax