Device Placement Optimization with Reinforcement Learning

Azalia Mirhoseini, Hieu Pham, Quoc V. Le +7

The past few years have witnessed a growth in size and computational requirements for training and inference with neural networks. Currently, a common approach to address these req…