torch named tensors API (Dimname etc.) incompatible with torch >= 2.13
Open
- Dominant language
- Python
- Stars
- 243
- Forks
- 38
- Avg merge
- 15h 32m
- Merged PRs (30d)
- 2
Description
PyTorch 2.13.0, released 2026-07-08, removed named tensors (PR #173895), deleting the Dimname overloads , including aten.prod.dim_Dimname.
torchax's amp.py still references that overload in a module-level CastPolicy dict at import time, so importing torchax with a newer PyTorch version crashes.
https://github.com/pytorch/pytorch/pull/173895
Contributor guide
Assessment
This issue has not been assessed yet.