Adding to kornia?

Hi,

This is a great work! Would you consider adding your model to kornia? https://github.com/kornia/kornia
I believe, that it could boost the usage and citations for you and make life a bit simpler for your users.

--
Best, Dmytro