MagmaDNN: Accelerated Deep Learning Using MAGMA Daniel Nichols Kwai Wong Stan Tomov
[email protected] [email protected] [email protected] University of Tennessee, Knoxville University of Tennessee, Knoxville University of Tennessee, Knoxville Lucien Ng Sihan Chen Alex Gessinger
[email protected] [email protected] [email protected] The Chinese University of Hong Kong The Chinese University of Hong Kong Slippery Rock University ABSTRACT 1 INTRODUCTION MagmaDNN [17] is a deep learning framework driven using the Machine Learning is becoming an increasingly vital aspect of to- highly optimized MAGMA dense linear algebra package. The library day’s technology, yet traditional techniques prove insucient for oers comparable performance to other popular frameworks, such complex problems [9]. Thus, Deep Learning (DL) is introduced, as TensorFlow, PyTorch, and Theano. C++ is used to implement the which is a methodology for learning data with multiple levels of framework providing fast memory operations, direct cuda access, abstraction [15]. DL is driven by Deep Neural Networks (DNN), and compile time errors. Common neural network layers such as which are Articial Neural Networks (ANN) comprised of multiple Fully Connected, Convolutional, Pooling, Flatten, and Dropout are layers and often convolutions and recurrence. DNNs are used to included. Hyperparameter tuning is performed with a parallel grid model complex data in numerous elds such as autonomous driving search engine. [5], handwriting recognition [16], image classication [14], speech- MagmaDNN uses several techniques to accelerate network train- to-text algorithms [11], and playing computationally dicult games ing. For instance, convolutions are performed using the Winograd [18]. algorithm and FFTs.