### 🚀 The feature, motivation and pitch follow pytorch pr 153666 ### Alternatives _No response_ ### Additional context _No response_