Get our free extension to see links to code for papers anywhere online!

Chrome logo  Add to Chrome

Firefox logo Add to Firefox

Picture for Zachary DeVito

Zachary DeVito

Facebook AI

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

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

* 14 pages, 8 figures, Submitted to MLSys 2022 

   Access Paper or Ask Questions

Using Python for Model Inference in Deep Learning

Zachary DeVito , Jason Ansel , Will Constable , Michael Suo , Ailing Zhang , Kim Hazelwood

   Access Paper or Ask Questions

Tensor Comprehensions: Framework-Agnostic High-Performance Machine Learning Abstractions

Nicolas Vasilache , Oleksandr Zinenko , Theodoros Theodoridis , Priya Goyal , Zachary DeVito , William S. Moses , Sven Verdoolaege , Andrew Adams , Albert Cohen

   Access Paper or Ask Questions

Opt: A Domain Specific Language for Non-linear Least Squares Optimization in Graphics and Imaging

Zachary DeVito , Michael Mara , Michael Zollhöfer , Gilbert Bernstein , Jonathan Ragan-Kelley , Christian Theobalt , Pat Hanrahan , Matthew Fisher , Matthias Nießner

   Access Paper or Ask Questions