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