research
∙
08/07/2023
Studying Large Language Model Generalization with Influence Functions
When trying to gain better visibility into a machine learning model in o...
share
research
∙
02/03/2023
Searching Large Neighborhoods for Integer Linear Programs with Contrastive Learning
Integer Linear Programs (ILPs) are powerful tools for modeling and solvi...
research
∙
01/09/2023
Learning to compile smartly for program size reduction
Compiler optimization passes are an important tool for improving program...
research
∙
12/15/2022
Local Branching Relaxation Heuristics for Integer Linear Programs
Large Neighborhood Search (LNS) is a popular heuristic algorithm for sol...
research
∙
10/24/2022
OLLA: Decreasing the Memory Usage of Neural Networks by Optimizing the Lifetime and Location of Arrays
The size of deep neural networks has grown exponentially in recent years...
research
∙
10/22/2022
SurCo: Learning Linear Surrogates For Combinatorial Nonlinear Optimization Problems
Optimization problems with expensive nonlinear cost functions and combin...
research
∙
05/02/2022
LoopStack: a Lightweight Tensor Algebra Compiler Stack
We present LoopStack, a domain specific compiler stack for tensor operat...
research
∙
01/29/2022
Flashlight: Enabling Innovation in Tools for Machine Learning
As the computational requirements for machine learning systems and the s...
research
∙
09/17/2021
CompilerGym: Robust, Performant Compiler Optimization Environments for AI Research
Interest in applying Artificial Intelligence (AI) techniques to compiler...
research
∙
08/27/2021
Using Graph Neural Networks to model the performance of Deep Neural Networks
With the unprecedented proliferation of machine learning software, there...
research
∙
06/19/2021
Learning Space Partitions for Path Planning
Path planning, the problem of efficiently discovering high-reward trajec...
research
∙
11/30/2020
Value Function Based Performance Optimization of Deep Learning Workloads
As machine learning techniques become ubiquitous, the efficiency of neur...
research
∙
12/03/2019
PyTorch: An Imperative Style, High-Performance Deep Learning Library
Deep learning frameworks have often focused on either usability or speed...
research
∙
06/13/2017
Device Placement Optimization with Reinforcement Learning
The past few years have witnessed a growth in size and computational req...
research
∙
03/14/2016