numpy
pycuda
tensorrt