AI RESEARCH

GraphMend: Code Transformations for Fixing Graph Breaks in PyTorch 2

arXiv CS.LG

ArXi:2509.16248v3 Announce Type: replace-cross This paper presents GRAPHMEND, a high-level compiler technique that eliminates FX graph breaks in PyTorch 2 programs. Although PyTorch 2