poc for using cuda torch from python :-) - View it on GitHub