Future Pytorch Implementation

[haoranD]

Dear Author,

Many thanks for such amazing works. I am very interested in your code.

Do you have any plan to release the Pytorch Version in the future?

Thanks again.

[yang-song]

We don't have plans to port this code to PyTorch in the short term. But we welcome any contributions on that!