Tensor Program Optimization with Probabilistic Programs

Tensor Program Optimization with Probabilistic Programs

26 May 2022

Masahiro Masuda

Papers citing "Tensor Program Optimization with Probabilistic Programs"

13 / 13 papers shown

Title
Improving Assembly Code Performance with Large Language Models via Reinforcement Learning Anjiang Wei Tarun Suresh Huanmi Tan Yinglun Xu Gagandeep Singh Ke Wang Alex Aiken 7 0 0 16 May 2025
QiMeng-TensorOp: Automatically Generating High-Performance Tensor Operators with Hardware Primitives X. Zhang Shaohui Peng Qirui Zhou Yuanbo Wen Qi Guo ... Ke Gao Chen Zhao Yanjun Wu Yunji Chen Ling Li VLM 39 0 0 08 May 2025
Tilus: A Virtual Machine for Arbitrary Low-Precision GPGPU Computation in LLM Serving Yaoyao Ding Bohan Hou X. Zhang Allan Lin Tianqi Chen Cody Yu Hao Yida Wang Gennady Pekhimenko 50 0 0 17 Apr 2025
PyGraph: Robust Compiler Support for CUDA Graphs in PyTorch Abhishek Ghosh Ajay Nayak Ashish Panwar Arkaprava Basu GNN 45 0 0 25 Mar 2025
FTuner: A Fast Dynamic Shape Tensors Program Auto-Tuner for Deep Learning Compilers Pengyu Mu Linquan Wei Yi Liu Rui Wang 25 0 0 31 Jul 2024
Optimal Kernel Orchestration for Tensor Programs with Korch Muyan Hu Ashwin Venkatram Shreyashri Biswas Balamurugan Marimuthu Bohan Hou Gabriele Oliaro Haojie Wang Liyan Zheng Xupeng Miao Jidong Zhai 120 4 0 13 Jun 2024
Allo: A Programming Model for Composable Accelerator Design Hongzheng Chen Niansong Zhang Shaojie Xiang Zhichen Zeng Mengjia Dai Zhiru Zhang 54 14 0 07 Apr 2024
DLAS: An Exploration and Assessment of the Deep Learning Acceleration Stack Perry Gibson José Cano Elliot J. Crowley Amos Storkey Michael F. P. O'Boyle 19 1 0 15 Nov 2023
Relax: Composable Abstractions for End-to-End Dynamic Machine Learning Ruihang Lai Junru Shao Siyuan Feng Steven Lyubomirsky Bohan Hou ... Sunghyun Park Prakalp Srivastava Jared Roesch T. Mowry Tianqi Chen 47 9 0 01 Nov 2023
Hidet: Task-Mapping Programming Paradigm for Deep Learning Tensor Programs Yaoyao Ding Cody Hao Yu Bojian Zheng Yizhi Liu Yida Wang Gennady Pekhimenko 26 30 0 18 Oct 2022
SparseTIR: Composable Abstractions for Sparse Compilation in Deep Learning Zihao Ye Ruihang Lai Junru Shao Tianqi Chen Luis Ceze 76 91 0 11 Jul 2022
MetaTune: Meta-Learning Based Cost Model for Fast and Efficient Auto-tuning Frameworks Jaehun Ryu Hyojin Sung 54 16 0 08 Feb 2021
Chameleon: Adaptive Code Optimization for Expedited Deep Neural Network Compilation Byung Hoon Ahn Prannoy Pilligundla Amir Yazdanbakhsh H. Esmaeilzadeh ODL 61 80 0 23 Jan 2020