Impelement a python interface in anticipation of cudNN suport