@article{torchfxpracticalprogramcaptureand,
  title = {Torch.fx: Practical Program Capture and Transformation for Deep Learning in Python},
  author = {James K. Reed and Zachary DeVito and Horace He and Ansley Ussery and Jason Ansel},
  year = {2021},
  eprint = {2112.08429},
  archivePrefix = {arXiv},
  url = {https://arxiv.org/abs/2112.08429v2},
}