Learning Generalizable Device Placement Algorithms for Distributed Machine Learning

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

摘要

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