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 Jason Ansel

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

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

* 14 pages, 8 figures, Submitted to MLSys 2022 

  Access Paper or Ask Questions

CompilerGym: Robust, Performant Compiler Optimization Environments for AI Research

Sep 17, 2021
Chris Cummins, Bram Wasti, Jiadong Guo, Brandon Cui, Jason Ansel, Sahir Gomez, Somya Jain, Jia Liu, Olivier Teytaud, Benoit Steiner, Yuandong Tian, Hugh Leather

* 12 pages. Source code available at 

  Access Paper or Ask Questions

Using Python for Model Inference in Deep Learning

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

  Access Paper or Ask Questions

Tight Prediction Intervals Using Expanded Interval Minimization

Jun 28, 2018
Dongqi Su, Ying Yin Ting, Jason Ansel

  Access Paper or Ask Questions