### ✨ Short description of the feature [tl;dr] Compatibility with pytorch 2.5 ### 💬 Detailed motivation and codes Compatibility with pytorch 2.5