Learning to Optimize Tensor Programs

NeurIPS 2018 Tianqi ChenLianmin ZhengEddie YanZiheng JiangThierry MoreauLuis CezeCarlos GuestrinArvind Krishnamurthy

We introduce a learning-based framework to optimize tensor programs for deep learning workloads. Efficient implementations of tensor operators, such as matrix multiplication and high dimensional convolution, are key enablers of effective deep learning systems... (read more)

PDF Abstract


No code implementations yet. Submit your code now


Results from the Paper

  Submit results from this paper to get state-of-the-art GitHub badges and help the community compare results to other papers.