Torch.fx: Practical Program Capture and Transformation for Deep Learning in Python

James K. Reed, Zachary DeVito, Horace He +2

Modern deep learning frameworks provide imperative, eager execution programming interfaces embedded in Python to provide a productive development experience. However, deep learning…