activity
20132019
most citedLearning to Optimize Tensor Programs

187 citations · 349 across the 7 of their papers we have counts for

collaborators

10 papers

cs.LG2019

Relay: A High-Level Compiler for Deep Learning

Jared Roesch, Steven Lyubomirsky, Marisa Kirisame +7

Frameworks for writing, compiling, and optimizing deep learning (DL) models have recently enabled progress in areas like computer vision and natural language processing. Extending…

cs.LG2018★ 7 cited

Automating Generation of Low Precision Deep Learning Operators

Meghan Cowan, Thierry Moreau, Tianqi Chen +1

State of the art deep learning models have made steady progress in the fields of computer vision and natural language processing, at the expense of growing model sizes and computat…

cs.AR2018★ 15 cited

Exploiting Errors for Efficiency: A Survey from Circuits to Algorithms

Phillip Stanley-Marbell, Armin Alaghi, Michael Carbin +13

When a computational task tolerates a relaxation of its specification or when an algorithm tolerates the effects of noise in its execution, hardware, programming languages, and sys…

cs.LG2018

A Hardware-Software Blueprint for Flexible Deep Learning Specialization

Thierry Moreau, Tianqi Chen, Luis Vega +8

Specialized Deep Learning (DL) acceleration stacks, designed for a specific set of frameworks, model architectures, operators, and data types, offer the allure of high performance…

cs.LG2018★ 187 cited

Learning to Optimize Tensor Programs

Tianqi Chen, Lianmin Zheng, Eddie Yan +5

We introduce a learning-based framework to optimize tensor programs for deep learning workloads. Efficient implementations of tensor operators, such as matrix multiplication and hi…

cs.LG2018★ 135 cited

TVM: An Automated End-to-End Optimizing Compiler for Deep Learning

Tianqi Chen, Thierry Moreau, Ziheng Jiang +9

There is an increasing need to bring machine learning to a wide diversity of hardware devices. Current frameworks rely on vendor-specific operator libraries and optimize for a narr…