Request for PyTorch 2.0 Compatibility

[wileewang]

First of all, thank you for creating and maintaining this amazing project. I have been using it for a while and it has been really helpful.

I noticed that the current version of the project seems to be compatible with older versions of PyTorch, and I would like to inquire if there are any plans to update the code to support PyTorch 2.0. The latest PyTorch version has introduced several new features and improvements, which could potentially benefit this project as well.

If you are already working on this update or plan to do so in the near future, I would be happy to know the estimated timeline. In case you need any assistance with testing or adapting the code, please let me know. I would be more than happy to contribute and help in any way possible.

Thank you once again for your great work and looking forward to the possibility of using this project with PyTorch 2.0.

Best regards