Benoit Steiner

发表

TensorFlow: A system for large-scale machine learning

Yuan Yu, Zhifeng Chen, Jianmin Chen, 2016, OSDI.

Device Placement Optimization with Reinforcement Learning

Samy Bengio, Naveen Kumar, Quoc V. Le, 2017, ICML.

A Hierarchical Model for Device Placement

Quoc V. Le, Azalia Mirhoseini, Hieu Pham, 2018, ICLR.

TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systems pdf

Martín Abadi, Martin Wattenberg, Michael Isard, 2016, ArXiv.

PyTorch: An Imperative Style, High-Performance Deep Learning Library

Natalia Gimelshein, Alykhan Tejani, Soumith Chintala, 2019, NeurIPS.

TensorFlow Distributions Joshua

Andy Davis, Geoffrey Irving, Dandelion Mané, 2017 .

ZiMM: a deep learning model for long term adverse events with non-clinical claims data

Ilya, Geoffrey Irving, Dandelion Mané, 2019 .

Hierarchical Planning for Device Placement

Quoc V. Le, Azalia Mirhoseini, Hieu Pham, 2018 .

Learning to optimize halide with tree search and random programs

Frédo Durand, Andrew Adams, Tzu-Mao Li, 2019, ACM Trans. Graph..

A graph neural network-based performance model for deep learning applications

Benoit Steiner, H. Leather, James Hegarty, 2022, Proceedings of the 6th ACM SIGPLAN International Symposium on Machine Programming.

LoopStack: a Lightweight Tensor Algebra Compiler Stack

Benoit Steiner, A. Zlateski, H. Leather, 2022, ArXiv.

Flashlight: Enabling Innovation in Tools for Machine Learning

Ronan Collobert, Benoit Steiner, Edouard Grave, 2022, ICML.

Neural Guidance for SAT Solving

Andy Davis, Geoffrey Irving, Yangqing Jia, 2018 .

Value Learning for Throughput Optimization of Deep Learning Workloads

Benoit Steiner, H. Leather, Chris Cummins, 2021, MLSys.

SurCo: Learning Linear Surrogates For Combinatorial Nonlinear Optimization Problems

Benoit Steiner, M. Schubert, B. Dilkina, 2022, ArXiv.

OLLA: Decreasing the Memory Usage of Neural Networks by Optimizing the Lifetime and Location of Arrays

Benoit Steiner, Jacob Kahn, James Hegarty, 2022, ArXiv.

OLLA: Optimizing the Lifetime and Location of Arrays to Reduce the Memory Usage of Neural Networks

Benoit Steiner, Jacob Kahn, James Hegarty, 2022, 2210.12924.

Learning Space Partitions for Path Planning pdf

Dan Klein, Joseph E. Gonzalez, Tianjun Zhang, 2021, NeurIPS.

Searching Large Neighborhoods for Integer Linear Programs with Contrastive Learning

Benoit Steiner, B. Dilkina, Yuandong Tian, 2023, ICML.

Local Branching Relaxation Heuristics for Integer Linear Programs

Benoit Steiner, B. Dilkina, Yuandong Tian, 2022, CPAIOR.

Ju n 20 17 Device Placement Optimization with Reinforcement Learning

Quoc V. Le, J. Dean, Samy Bengio, 2017 .

A H IERARCHICAL M ODEL FOR D EVICE P LACEMENT

Benoit Steiner, Azalia Mirhoseini, Hieu Pham, 2018 .

Learning Compiler Pass Orders using Coreset and Normalized Value Prediction

P. Xie, Benoit Steiner, H. Leather, 2023, ICML.

SurCo: Learning Linear Surrogates For Combinatorial Nonlinear Optimization Problems

Benoit Steiner, M. Schubert, B. Dilkina, 2022, ICML.

CompilerGym: Robust, Performant Compiler Optimization Environments for AI Research pdf

Olivier Teytaud, Brandon Cui, Hugh Leather, 2021, 2022 IEEE/ACM International Symposium on Code Generation and Optimization (CGO).

VALUE LEARNING FOR THROUGHPUT OPTIMIZATION OF DEEP NEURAL NETWORKS

Benoit Steiner, H. Leather, Chris Cummins, 2021 .

Studying Large Language Model Generalization with Influence Functions

Benoit Steiner, Sam McCandlish, Esin Durmus, 2023, ArXiv.

Measuring Faithfulness in Chain-of-Thought Reasoning

Benoit Steiner, T. Henighan, Esin Durmus, 2023, ArXiv.

Using Graph Neural Networks to model the performance of Deep Neural Networks pdf

Hugh Leather, Benoit Steiner, James Hegarty, 2021, ArXiv.

TensorFlow: A system for large-scale machine learning

Andy Davis, J. Dean, M. Devin, 2016, OSDI.

Learning to compile smartly for program size reduction

P. Xie, Benoit Steiner, Chris Cummins, 2023 .