1 code implementation • NeurIPS 2019 • 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.
3 code implementations • 20 Jun 2019 • Ravichandra Addanki, Shaileshh Bojja Venkatakrishnan, Shreyan Gupta, Hongzi Mao, Mohammad Alizadeh
Unlike prior approaches that only find a device placement for a specific computation graph, Placeto can learn generalizable device placement policies that can be applied to any graph.