As a follow-up to #1537, this adds type annotations for torch. This is the most difficult step so far
pykeen
