gpu version of numba compiler