Implementation of CUDA using python