Target-independent XLA optimization using Reinforcement Learning

Target-independent XLA optimization using Reinforcement Learning

28 August 2023

ArXiv (abs)PDF HTML

Papers citing "Target-independent XLA optimization using Reinforcement Learning"

17 / 17 papers shown

Title
A2C is a special case of PPO Shengyi Huang Anssi Kanervisto Antonin Raffin Weixun Wang Santiago Ontañón Rousslan Fernand Julien Dossa OffRL 103 26 0 18 May 2022
CompilerGym: Robust, Performant Compiler Optimization Environments for AI Research Chris Cummins Bram Wasti Jiadong Guo Brandon Cui Jason Ansel ... Jia-Wei Liu O. Teytaud Benoit Steiner Yuandong Tian Hugh Leather 63 74 0 17 Sep 2021
MLGO: a Machine Learning Guided Compiler Optimizations Framework Mircea Trofin Yundi Qian E. Brevdo Zinan Lin K. Choromanski Didong Li 75 66 0 13 Jan 2021
Transferable Graph Optimizers for ML Compilers Yanqi Zhou Sudip Roy AmirAli Abdolrashidi Daniel Wong Peter C. Ma ... Mangpo Phitchaya Phothilimtha Shen Wang Anna Goldie Azalia Mirhoseini James Laudon GNN 53 55 0 21 Oct 2020
Beyond variance reduction: Understanding the true impact of baselines on policy optimization Wesley Chung Valentin Thomas Marlos C. Machado Nicolas Le Roux OffRL 92 24 0 31 Aug 2020
Static Neural Compiler Optimization via Deep Reinforcement Learning Rahim Mammadli Ali Jannesari F. Wolf 53 31 0 20 Aug 2020
Ansor: Generating High-Performance Tensor Programs for Deep Learning Lianmin Zheng Chengfan Jia Minmin Sun Zhao Wu Cody Hao Yu ... Jun Yang Danyang Zhuo Koushik Sen Joseph E. Gonzalez Ion Stoica 144 403 0 11 Jun 2020
A Collaborative Filtering Approach for the Automatic Tuning of Compiler Optimisations Stefano Cereda G. Palermo Paolo Cremonesi Stefano Doni 29 13 0 06 May 2020
Chameleon: Adaptive Code Optimization for Expedited Deep Neural Network Compilation Byung Hoon Ahn Prannoy Pilligundla Amir Yazdanbakhsh H. Esmaeilzadeh ODL 100 82 0 23 Jan 2020
NeuroVectorizer: End-to-End Vectorization with Deep Reinforcement Learning Ameer Haj-Ali Nesreen Ahmed Theodore L. Willke Sophia Shao Krste Asanović Ion Stoica 61 101 0 20 Sep 2019
Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelism Mohammad Shoeybi M. Patwary Raul Puri P. LeGresley Jared Casper Bryan Catanzaro MoE 339 1,918 0 17 Sep 2019
Learning to Optimize Tensor Programs Tianqi Chen Lianmin Zheng Eddie Q. Yan Ziheng Jiang T. Moreau Luis Ceze Carlos Guestrin Arvind Krishnamurthy 89 404 0 21 May 2018
Tensor Comprehensions: Framework-Agnostic High-Performance Machine Learning Abstractions Nicolas Vasilache O. Zinenko Theodoros Theodoridis Priya Goyal Zach DeVito William S. Moses Sven Verdoolaege Andrew Adams Albert Cohen 76 436 0 13 Feb 2018
A Survey on Compiler Autotuning using Machine Learning Amir H. Ashouri W. Killian John Cavazos G. Palermo Cristina Silvano 88 203 0 13 Jan 2018
Proximal Policy Optimization Algorithms John Schulman Filip Wolski Prafulla Dhariwal Alec Radford Oleg Klimov OffRL 568 19,296 0 20 Jul 2017
OpenAI Gym Greg Brockman Vicki Cheung Ludwig Pettersson Jonas Schneider John Schulman Jie Tang Wojciech Zaremba OffRL ODL 225 5,087 0 05 Jun 2016
Asynchronous Methods for Deep Reinforcement Learning Volodymyr Mnih Adria Puigdomenech Badia M. Berk Mirza Alex Graves Timothy Lillicrap Tim Harley David Silver Koray Kavukcuoglu 210 8,881 0 04 Feb 2016