Work with newer torch

[fecet]

I cannot install this as torch 1.4.0 doesn't exist at pypi, have you tested that for new torch, say torch 1.13 or even torch 2.0?