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 https://github.com/facebookresearch/CompilerGym 

  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