GPU- Raytracer in python