The paper presents a parallel math library, dMath, that demonstrates leading scaling when using intranode, internode, and hybrid-parallelism for deep learning (DL). dMath provides easy-to-use distributed primitives and a variety of domain-specific algorithms including matrix multiplication, convolutions, and others allowing for rapid development of scalable applications like deep neural networks (DNNs). Persistent data stored in GPU memory and advanced memory management techniques avoid costly transfers between host and device. dMath delivers performance, portability, and productivity to its specific domain of support.
@article{arxiv.1611.07819,
title = {dMath: Distributed Linear Algebra for DL},
author = {Steven Eliuk and Cameron Upright and Hars Vardhan and Stephen Walsh and Trevor Gale},
journal= {arXiv preprint arXiv:1611.07819},
year = {2016}
}
Comments
5 pages. arXiv admin note: text overlap with arXiv:1604.01416