Comment by pancho111203

Cool stuff, but I would have found it more useful as a python library, to allow integration with existing tools like Pytorch.