numba for gpu kernels... cool!