Description
🚀 The feature, motivation and pitch
We should remove redundant type aliases for _device_t and replace with torch.types.Device where appropriate to make the typing system a bit more consistent.
#152935 is a good step in that direction
Alternatives
No response
Additional context
No response