numba
numpy
pycuda
tensorflow-gpu
