A distributed data-parallel pytorch implementation of the distributed shampoo optimizer for training neural networks at-scale

Hao-Jun Michael Shi, Tsung-Hsien Lee, Shintaro Iwasaki, Jose Gallego-Posada, Zhijing Li, Kaushik Rangadurai, Dheevatsa Mudigere, Michael Rabbat · 2023

1 Pith paper cite this work. Polarity classification is still indexing.

1 Pith paper citing it