Learning Generalizable Device Placement Algorithms for Distributed Machine Learning

Authors: ravichandra addanki, Shaileshh Bojja Venkatakrishnan, Shreyan Gupta, Hongzi Mao, Mohammad Alizadeh

Abstract

We present Placeto, a reinforcement learning (RL) approach to efficiently find device placements for distributed neural network training.