Roman Garipov's gpu_parallel.py from hogwild_inference - View it on GitHub