Search Results for author: Jason Ansel

Found 4 papers, 1 papers with code

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

no code implementations15 Dec 2021 James K. Reed, Zachary DeVito, Horace He, Ansley Ussery, Jason Ansel

Modern deep learning frameworks provide imperative, eager execution programming interfaces embedded in Python to provide a productive development experience.

CompilerGym: Robust, Performant Compiler Optimization Environments for AI Research

1 code implementation17 Sep 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

What is needed is an easy, reusable experimental infrastructure for real world compiler optimization tasks that can serve as a common benchmark for comparing techniques, and as a platform to accelerate progress in the field.

Compiler Optimization OpenAI Gym

Using Python for Model Inference in Deep Learning

no code implementations1 Apr 2021 Zachary DeVito, Jason Ansel, Will Constable, Michael Suo, Ailing Zhang, Kim Hazelwood

We evaluate our design on a suite of popular PyTorch models on Github, showing how they can be packaged in our inference format, and comparing their performance to TorchScript.

Model extraction

Tight Prediction Intervals Using Expanded Interval Minimization

no code implementations28 Jun 2018 Dongqi Su, Ying Yin Ting, Jason Ansel

Many prior techniques for generating prediction intervals make assumptions on the distribution of error, which causes them to work poorly for problems with asymmetric distributions.

Prediction Intervals

Cannot find the paper you are looking for? You can Submit a new open access paper.