Software-hardware co-design for fast and scalable training of deep learning recommendation models
Author:
Mudigere Dheevatsa1, Hao Yuchen1, Huang Jianyu1, Jia Zhihao2, Tulloch Andrew1, Sridharan Srinivas1, Liu Xing1, Ozdal Mustafa1, Nie Jade1, Park Jongsoo1, Luo Liang1, Yang Jie (Amy)1, Gao Leon1, Ivchenko Dmytro1, Basant Aarti1, Hu Yuxi1, Yang Jiyan1, Ardestani Ehsan K.1, Wang Xiaodong1, Komuravelli Rakesh1, Chu Ching-Hsiang1, Yilmaz Serhat1, Li Huayu1, Qian Jiyuan1, Feng Zhuobo1, Ma Yinbin1, Yang Junjie1, Wen Ellie1, Li Hong1, Yang Lin1, Sun Chonglin1, Zhao Whitney1, Melts Dimitry1, Dhulipala Krishna1, Kishore KR1, Graf Tyler1, Eisenman Assaf1, Matam Kiran Kumar1, Gangidi Adi1, Chen Guoqiang Jerry1, Krishnan Manoj1, Nayak Avinash1, Nair Krishnakumar1, Muthiah Bharath1, khorashadi Mahmoud1, Bhattacharya Pallab1, Lapukhov Petr1, Naumov Maxim1, Mathews Ajit1, Qiao Lin1, Smelyanskiy Mikhail1, Jia Bill1, Rao Vijay1
Affiliation:
1. Meta Platforms 2. Carnegie Mellon University
Reference71 articles.
1. [n.d.]. NVIDIA Collective Communications Library (NCCL) https://developer.nvidia.com/nccl. [n.d.]. NVIDIA Collective Communications Library (NCCL) https://developer.nvidia.com/nccl. 2. [n.d.]. OCP Open rack standard (v2) https://www.opencompute.org/wiki/Open_Rack/SpecsAndDesigns#RACK_Standards. [n.d.]. OCP Open rack standard (v2) https://www.opencompute.org/wiki/Open_Rack/SpecsAndDesigns#RACK_Standards. 3. Martín Abadi Ashish Agarwal Paul Barham Eugene Brevdo Zhifeng Chen Craig Citro Greg S. Corrado Andy Davis Jeffrey Dean Matthieu Devin Sanjay Ghemawat Ian Goodfellow Andrew Harp Geoffrey Irving Michael Isard Yangqing Jia Rafal Jozefowicz Lukasz Kaiser Manjunath Kudlur Josh Levenberg Dandelion Mané Rajat Monga Sherry Moore Derek Murray Chris Olah Mike Schuster Jonathon Shlens Benoit Steiner Ilya Sutskever Kunal Talwar Paul Tucker Vincent Vanhoucke Vijay Vasudevan Fernanda Viégas Oriol Vinyals Pete Warden Martin Wattenberg Martin Wicke Yuan Yu and Xiaoqiang Zheng. 2015. TensorFlow: Large-Scale Machine Learning on Heterogeneous Systems. https://www.tensorflow.org/ Software available from tensorflow.org. Martín Abadi Ashish Agarwal Paul Barham Eugene Brevdo Zhifeng Chen Craig Citro Greg S. Corrado Andy Davis Jeffrey Dean Matthieu Devin Sanjay Ghemawat Ian Goodfellow Andrew Harp Geoffrey Irving Michael Isard Yangqing Jia Rafal Jozefowicz Lukasz Kaiser Manjunath Kudlur Josh Levenberg Dandelion Mané Rajat Monga Sherry Moore Derek Murray Chris Olah Mike Schuster Jonathon Shlens Benoit Steiner Ilya Sutskever Kunal Talwar Paul Tucker Vincent Vanhoucke Vijay Vasudevan Fernanda Viégas Oriol Vinyals Pete Warden Martin Wattenberg Martin Wicke Yuan Yu and Xiaoqiang Zheng. 2015. TensorFlow: Large-Scale Machine Learning on Heterogeneous Systems. https://www.tensorflow.org/ Software available from tensorflow.org. 4. Bilge Acun Matthew Murphy Xiaodong Wang Jade Nie Carole-Jean Wu and Kim Hazelwood. 2020. Understanding Training Efficiency of Deep Learning Recommendation Models at Scale. arXiv:2011.05497 [cs.AR] Bilge Acun Matthew Murphy Xiaodong Wang Jade Nie Carole-Jean Wu and Kim Hazelwood. 2020. Understanding Training Efficiency of Deep Learning Recommendation Models at Scale. arXiv:2011.05497 [cs.AR] 5. Tom B. Brown Benjamin Mann Nick Ryder Melanie Subbiah Jared Kaplan Prafulla Dhariwal Arvind Neelakantan Pranav Shyam Girish Sastry Amanda Askell Sandhini Agarwal Ariel Herbert-Voss Gretchen Krueger Tom Henighan Rewon Child Aditya Ramesh Daniel M. Ziegler Jeffrey Wu Clemens Winter Christopher Hesse Mark Chen Eric Sigler Mateusz Litwin Scott Gray Benjamin Chess Jack Clark Christopher Berner Sam McCandlish Alec Radford Ilya Sutskever and Dario Amodei. 2020. Language Models are Few-Shot Learners. arXiv:2005.14165 [cs.CL] Tom B. Brown Benjamin Mann Nick Ryder Melanie Subbiah Jared Kaplan Prafulla Dhariwal Arvind Neelakantan Pranav Shyam Girish Sastry Amanda Askell Sandhini Agarwal Ariel Herbert-Voss Gretchen Krueger Tom Henighan Rewon Child Aditya Ramesh Daniel M. Ziegler Jeffrey Wu Clemens Winter Christopher Hesse Mark Chen Eric Sigler Mateusz Litwin Scott Gray Benjamin Chess Jack Clark Christopher Berner Sam McCandlish Alec Radford Ilya Sutskever and Dario Amodei. 2020. Language Models are Few-Shot Learners. arXiv:2005.14165 [cs.CL]
Cited by
56 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献
|
|