Search Results for author: Pengming Wang

Found 3 papers, 2 papers with code

Optimizing Memory Mapping Using Deep Reinforcement Learning

no code implementations11 May 2023 Pengming Wang, Mikita Sazanovich, Berkin Ilbeyi, Phitchaya Mangpo Phothilimthana, Manish Purohit, Han Yang Tay, Ngân Vũ, Miaosen Wang, Cosmin Paduraru, Edouard Leurent, Anton Zhernov, Po-Sen Huang, Julian Schrittwieser, Thomas Hubert, Robert Tung, Paula Kurylowicz, Kieran Milan, Oriol Vinyals, Daniel J. Mankowitz

We also introduce a Reinforcement Learning agent, mallocMuZero, and show that it is capable of playing this game to discover new and improved memory mapping solutions that lead to faster execution times on real ML workloads on ML accelerators.

Cloud Computing Decision Making +3

Learning a Large Neighborhood Search Algorithm for Mixed Integer Programs

1 code implementation21 Jul 2021 Nicolas Sonnerat, Pengming Wang, Ira Ktena, Sergey Bartunov, Vinod Nair

Large Neighborhood Search (LNS) is a combinatorial optimization heuristic that starts with an assignment of values for the variables to be optimized, and iteratively improves it by searching a large neighborhood around the current assignment.

Combinatorial Optimization Imitation Learning

Cannot find the paper you are looking for? You can Submit a new open access paper.