arxiv.org
A Data-Centric Optimization Framework for Machine Learning
October 2021 • Oliver Rausch, Tal Ben‐Nun, Nikoli Dryden, Андрей Иванов, Shigang Li, Torsten Hoefler
Rapid progress in deep learning is leading to a diverse set of quickly changing models, with a dramatically growing demand for compute. However, as frameworks specialize performance optimization to patterns in popular networks, they implicitly constrain novel and diverse models that drive progress in research. We empower deep learning researchers by defining a flexible and user-customizable pipeline for optimizing training of arbitrary deep neural networks, based on data movement minimization. The pipeline begins …