conftrace_

device placement

8 papers

Explore in graph

Co-occurring keywords

reinforcement learning (4352) neural network optimization (1293) neural network training (309) computation graph (10) computational graph (12) graph neural network (3962) proximal policy optimization (123) distributed training (188) neural network compilation (2) compiler optimization (29)

Papers

A Structure-Aware Framework for Learning Device Placements on Computation Graphs NIPS 2024
DevFormer: A Symmetric Transformer for Context-Aware Device Placement ICML 2023
DreamShard: Generalizable Embedding Table Placement for Recommender Systems NIPS 2022
Efficient Algorithms for Device Placement of DNN Graph Operators NIPS 2020
Transferable Graph Optimizers for ML Compilers NIPS 2020
Learning Generalizable Device Placement Algorithms for Distributed Machine Learning NIPS 2019
Post: Device Placement with Cross-Entropy Minimization and Proximal Policy Optimization NIPS 2018
Spotlight: Optimizing Device Placement for Training Deep Neural Networks ICML 2018