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

James K. Reed, Zachary DeVito, Horace He, Ansley Ussery, Jason Ansel