X-RLflow: Graph Reinforcement Learning for Neural Network Subgraphs Transformation

28 April 2023

Papers citing "X-RLflow: Graph Reinforcement Learning for Neural Network Subgraphs Transformation"

5 / 5 papers shown

Title
Optimizing Tensor Computation Graphs with Equality Saturation and Monte Carlo Tree Search Jakob Hartmann Guoliang He Eiko Yoneki 24 0 0 07 Oct 2024
A Reinforcement Learning Environment for Automatic Code Optimization in the MLIR Compiler Nazim Bendib Iheb Nassim Aouadj Riyadh Baghdadi 24 1 0 17 Sep 2024
PrefixRL: Optimization of Parallel Prefix Circuits using Deep Reinforcement Learning Rajarshi Roy Jonathan Raiman Neel Kant Ilyas Elkin Robert M. Kirby Michael Siu S. Oberman Saad Godil Bryan Catanzaro 35 38 0 14 May 2022
Zero-Shot Text-to-Image Generation Aditya A. Ramesh Mikhail Pavlov Gabriel Goh Scott Gray Chelsea Voss Alec Radford Mark Chen Ilya Sutskever VLM 255 4,796 0 24 Feb 2021
Equality Saturation for Tensor Graph Superoptimization Yichen Yang Mangpo Phitchaya Phothilimtha Y. Wang Max Willsey Sudip Roy Jacques Pienaar 38 81 0 05 Jan 2021