Just saw that PyTorch core made a branch cut for 2.5 release. Now nightly is 2.6. Probably a good idea to add CI to run on PyTorch 2.5rc? (don't even know if it's possible to install rc version of PyTorch). This would help to catch changes, especially new features, that for some reasons don't work with 2.5 but work with latest nightly.
Other miscs stuff:
- Add
TORCH_VERSION_AT_LEAST_2_6
- Remove official support and CI for PyTorch 2.2?