That is a pretty interessing read, it seen it require some manual compilation and a few changes, it would be great if pytorch could implement it officially, but I will take a look as well, thanks for that.